Skip to content

Commit 9092653

Browse files
committed
add bloom ipex inference test
1 parent 5120f75 commit 9092653

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

tests/ipex/test_inference.py

+11-2
Original file line numberDiff line numberDiff line change
@@ -33,9 +33,9 @@
3333

3434
MODEL_NAMES = {
3535
"bert": "hf-internal-testing/tiny-random-bert",
36+
"bloom": "hf-internal-testing/tiny-random-BloomModel",
3637
"distilbert": "hf-internal-testing/tiny-random-distilbert",
3738
"roberta": "hf-internal-testing/tiny-random-roberta",
38-
"bloom": "hf-internal-testing/tiny-random-bloom",
3939
"gptj": "hf-internal-testing/tiny-random-gptj",
4040
"gpt2": "hf-internal-testing/tiny-random-gpt2",
4141
"gpt_neo": "hf-internal-testing/tiny-random-GPTNeoModel",
@@ -59,7 +59,16 @@ class IPEXIntegrationTest(unittest.TestCase):
5959
"roberta",
6060
)
6161

62-
TEXT_GENERATION_SUPPORTED_ARCHITECTURES = ("gptj", "gpt2", "gpt_neo", "gpt_bigcode", "llama", "opt", "mpt")
62+
TEXT_GENERATION_SUPPORTED_ARCHITECTURES = (
63+
"bloom",
64+
"gptj",
65+
"gpt2",
66+
"gpt_neo",
67+
"gpt_bigcode",
68+
"llama",
69+
"opt",
70+
"mpt",
71+
)
6372

6473
QA_SUPPORTED_ARCHITECTURES = (
6574
"bert",

0 commit comments

Comments
 (0)