add bloom ipex inference test

huggingface · Feb 5, 2024 · 9092653 · 9092653
1 parent 5120f75
commit 9092653
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/tests/ipex/test_inference.py b/tests/ipex/test_inference.py
@@ -33,9 +33,9 @@
 
 MODEL_NAMES = {
     "bert": "hf-internal-testing/tiny-random-bert",
+    "bloom": "hf-internal-testing/tiny-random-BloomModel",
     "distilbert": "hf-internal-testing/tiny-random-distilbert",
     "roberta": "hf-internal-testing/tiny-random-roberta",
-    "bloom": "hf-internal-testing/tiny-random-bloom",
     "gptj": "hf-internal-testing/tiny-random-gptj",
     "gpt2": "hf-internal-testing/tiny-random-gpt2",
     "gpt_neo": "hf-internal-testing/tiny-random-GPTNeoModel",
@@ -59,7 +59,16 @@ class IPEXIntegrationTest(unittest.TestCase):
         "roberta",
     )
 
-    TEXT_GENERATION_SUPPORTED_ARCHITECTURES = ("gptj", "gpt2", "gpt_neo", "gpt_bigcode", "llama", "opt", "mpt")
+    TEXT_GENERATION_SUPPORTED_ARCHITECTURES = (
+        "bloom",
+        "gptj",
+        "gpt2",
+        "gpt_neo",
+        "gpt_bigcode",
+        "llama",
+        "opt",
+        "mpt",
+    )
 
     QA_SUPPORTED_ARCHITECTURES = (
         "bert",