Skip to content

Commit 88a4f52

Browse files
committed
add test model
1 parent 827bb9e commit 88a4f52

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

tests/openvino/test_modeling.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -535,12 +535,13 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
535535
"stablelm",
536536
"starcoder2",
537537
"phi",
538+
"phi3",
538539
"internlm2",
539540
"orion",
540541
"falcon",
541542
)
542543
GENERATION_LENGTH = 100
543-
REMOTE_CODE_MODELS = ("chatglm", "minicpm", "baichuan2", "jais", "qwen", "internlm2", "olmo", "orion")
544+
REMOTE_CODE_MODELS = ("chatglm", "minicpm", "baichuan2", "jais", "qwen", "internlm2", "olmo", "orion", "phi3")
544545

545546
@parameterized.expand(SUPPORTED_ARCHITECTURES)
546547
def test_compare_to_transformers(self, model_arch):

tests/openvino/utils_tests.py

+1
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,7 @@
7777
"pegasus": "hf-internal-testing/tiny-random-pegasus",
7878
"pix2struct": "fxmarty/pix2struct-tiny-random",
7979
"phi": "echarlaix/tiny-random-PhiForCausalLM",
80+
"phi3": "katuni4ka/tiny-random-phi3",
8081
"poolformer": "hf-internal-testing/tiny-random-PoolFormerModel",
8182
"qwen": "katuni4ka/tiny-random-qwen",
8283
"qwen2": "Qwen/Qwen1.5-0.5B",

0 commit comments

Comments
 (0)