Skip to content

Commit d68cf15

Browse files
committed
add test for baichuan 13b
1 parent 4c8cb0e commit d68cf15

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

tests/openvino/test_modeling.py

+12-1
Original file line numberDiff line numberDiff line change
@@ -510,6 +510,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
510510
SUPPORTED_ARCHITECTURES = (
511511
"bart",
512512
"baichuan2",
513+
"baichuan2-13b",
513514
"gpt_bigcode",
514515
"blenderbot",
515516
"blenderbot-small",
@@ -540,7 +541,17 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
540541
"falcon",
541542
)
542543
GENERATION_LENGTH = 100
543-
REMOTE_CODE_MODELS = ("chatglm", "minicpm", "baichuan2", "jais", "qwen", "internlm2", "olmo", "orion")
544+
REMOTE_CODE_MODELS = (
545+
"chatglm",
546+
"minicpm",
547+
"baichuan2",
548+
"baichuan2-13b",
549+
"jais",
550+
"qwen",
551+
"internlm2",
552+
"olmo",
553+
"orion",
554+
)
544555

545556
@parameterized.expand(SUPPORTED_ARCHITECTURES)
546557
def test_compare_to_transformers(self, model_arch):

0 commit comments

Comments
 (0)