add gptbigcode to ipex test

Signed-off-by: changwangss <[email protected]>
huggingface · Sep 26, 2023 · 183da95 · 183da95
1 parent e0c7486
commit 183da95
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/ipex/test_inference.py b/tests/ipex/test_inference.py
@@ -40,6 +40,7 @@
     "gpt2": "hf-internal-testing/tiny-random-gpt2",
     "gpt_neo": "hf-internal-testing/tiny-random-GPTNeoModel",
     "gpt_neox": "hf-internal-testing/tiny-random-GPTNeoXForCausalLM",
+    "gpt_bigcode": "hf-internal-testing/tiny-random-GPTBigCodeModel",
 }
 
 _CLASSIFICATION_TASK_TO_AUTOMODELS = {
@@ -59,6 +60,7 @@ class IPEXIntegrationTest(unittest.TestCase):
         "gptj",
         "gpt2",
         "gpt_neo",
+        "gpt_bigcode"
     )
 
     QA_SUPPORTED_ARCHITECTURES = (