Skip to content

Commit 98d3d48

Browse files
committed
set seed
1 parent 56b152e commit 98d3d48

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/openvino/test_modeling.py

+1
Original file line numberDiff line numberDiff line change
@@ -535,6 +535,7 @@ def test_compare_to_transformers(self, model_arch):
535535
self.assertEqual(
536536
ov_model.stateful, self.IS_SUPPORT_STATEFUL and ov_model.config.model_type not in not_stateful
537537
)
538+
set_seed(SEED)
538539
transformers_model = AutoModelForCausalLM.from_pretrained(model_id, **model_kwargs)
539540
tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=model_arch in self.REMOTE_CODE_MODELS)
540541
if model_arch == "qwen":

0 commit comments

Comments
 (0)