diff --git a/tests/ipex/test_inference.py b/tests/ipex/test_inference.py index d5cf571150..f1aa44468f 100644 --- 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 = (