Skip to content

Commit 1eeba6b

Browse files
committed
fix test for chatglm
1 parent 9fea0b3 commit 1eeba6b

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/openvino/test_modeling.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -592,8 +592,9 @@ def test_compare_to_transformers(self, model_arch):
592592
if model_arch == "qwen":
593593
return
594594

595-
# Compare batched generation.
596-
tokenizer.pad_token_id = tokenizer.eos_token_id
595+
if model_arch != "chatglm":
596+
tokenizer.pad_token_id = tokenizer.eos_token_id
597+
# Compare batched generation
597598
tokenizer.padding_side = "left"
598599
tokens = tokenizer(["Today is a nice day and I am longer", "This is me"], return_tensors="pt", padding=True)
599600
ov_model.generation_config.eos_token_id = None

0 commit comments

Comments
 (0)