From 183da95cf02d005bd9da2e1a403ca830569558fe Mon Sep 17 00:00:00 2001
From: changwangss <chang1.wang@intel.com>
Date: Mon, 25 Sep 2023 18:35:52 -0700
Subject: [PATCH] add gptbigcode to ipex test

Signed-off-by: changwangss <chang1.wang@intel.com>
---
 tests/ipex/test_inference.py | 2 ++
 1 file changed, 2 insertions(+)

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 = (