change llama test model to cover MQA

jiqing-feng · jiqing-feng · commit ea507987cfdd · 2024-03-05T08:21:13.000-05:00
diff --git a/tests/generation/test_modeling.py b/tests/generation/test_modeling.py
@@ -30,7 +30,7 @@
     "gpt2": "hf-internal-testing/tiny-random-gpt2",
     "gpt_neo": "hf-internal-testing/tiny-random-GPTNeoModel",
     "mistral": "echarlaix/tiny-random-mistral",
-    "llama": "fxmarty/tiny-llama-fast-tokenizer",
+    "llama": "Jiqing/tiny_random_llama2",
     "gpt_bigcode": "hf-internal-testing/tiny-random-GPTBigCodeModel",
 }
 
diff --git a/tests/ipex/test_inference.py b/tests/ipex/test_inference.py
@@ -41,7 +41,7 @@
     "gpt_neo": "hf-internal-testing/tiny-random-GPTNeoModel",
     "gpt_neox": "hf-internal-testing/tiny-random-GPTNeoXForCausalLM",
     "gpt_bigcode": "hf-internal-testing/tiny-random-GPTBigCodeModel",
-    "llama": "fxmarty/tiny-llama-fast-tokenizer",
+    "llama": "Jiqing/tiny_random_llama2",
     "opt": "hf-internal-testing/tiny-random-OPTModel",
     "mpt": "hf-internal-testing/tiny-random-MptForCausalLM",
 }
diff --git a/tests/ipex/test_modeling.py b/tests/ipex/test_modeling.py
@@ -66,7 +66,7 @@
     "gpt_neox": "hf-internal-testing/tiny-random-GPTNeoXForCausalLM",
     "gptj": "hf-internal-testing/tiny-random-GPTJModel",
     "levit": "hf-internal-testing/tiny-random-LevitModel",
-    "llama": "fxmarty/tiny-llama-fast-tokenizer",
+    "llama": "Jiqing/tiny_random_llama2",
     "marian": "sshleifer/tiny-marian-en-de",
     "mbart": "hf-internal-testing/tiny-random-mbart",
     "mistral": "echarlaix/tiny-random-mistral",

Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@`
`30`	`30`	`"gpt2": "hf-internal-testing/tiny-random-gpt2",`
`31`	`31`	`"gpt_neo": "hf-internal-testing/tiny-random-GPTNeoModel",`
`32`	`32`	`"mistral": "echarlaix/tiny-random-mistral",`
`33`		`- "llama": "fxmarty/tiny-llama-fast-tokenizer",`
	`33`	`+ "llama": "Jiqing/tiny_random_llama2",`
`34`	`34`	`"gpt_bigcode": "hf-internal-testing/tiny-random-GPTBigCodeModel",`
`35`	`35`	`}`
`36`	`36`
Original file line number	Diff line number	Diff line change
`@@ -41,7 +41,7 @@`
`41`	`41`	`"gpt_neo": "hf-internal-testing/tiny-random-GPTNeoModel",`
`42`	`42`	`"gpt_neox": "hf-internal-testing/tiny-random-GPTNeoXForCausalLM",`
`43`	`43`	`"gpt_bigcode": "hf-internal-testing/tiny-random-GPTBigCodeModel",`
`44`		`- "llama": "fxmarty/tiny-llama-fast-tokenizer",`
	`44`	`+ "llama": "Jiqing/tiny_random_llama2",`
`45`	`45`	`"opt": "hf-internal-testing/tiny-random-OPTModel",`
`46`	`46`	`"mpt": "hf-internal-testing/tiny-random-MptForCausalLM",`
`47`	`47`	`}`