set seed

echarlaix · echarlaix · commit 98d3d48092e8 · 2024-03-21T17:43:36.000+01:00
diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py
@@ -535,6 +535,7 @@ def test_compare_to_transformers(self, model_arch):
         self.assertEqual(
             ov_model.stateful, self.IS_SUPPORT_STATEFUL and ov_model.config.model_type not in not_stateful
         )
+        set_seed(SEED)
         transformers_model = AutoModelForCausalLM.from_pretrained(model_id, **model_kwargs)
         tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=model_arch in self.REMOTE_CODE_MODELS)
         if model_arch == "qwen":

Original file line number	Diff line number	Diff line change
`@@ -535,6 +535,7 @@ def test_compare_to_transformers(self, model_arch):`
`535`	`535`	`self.assertEqual(`
`536`	`536`	`ov_model.stateful, self.IS_SUPPORT_STATEFUL and ov_model.config.model_type not in not_stateful`
`537`	`537`	`)`
	`538`	`+ set_seed(SEED)`
`538`	`539`	`transformers_model = AutoModelForCausalLM.from_pretrained(model_id, **model_kwargs)`
`539`	`540`	`tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=model_arch in self.REMOTE_CODE_MODELS)`
`540`	`541`	`if model_arch == "qwen":`