Skip to content

Commit 5f9b1c4

Browse files
committed
fix
1 parent 2417f68 commit 5f9b1c4

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

tests/openvino/test_modeling.py

+2-4
Original file line numberDiff line numberDiff line change
@@ -524,10 +524,8 @@ def test_compare_to_transformers(self, model_arch):
524524

525525
model_kwargs = {}
526526
if model_arch in self.REMOTE_CODE_MODELS:
527-
model_kwargs = {
528-
"config": AutoConfig.from_pretrained(model_id, trust_remote_code=True),
529-
"trust_remote_code": True,
530-
}
527+
model_kwargs = {"trust_remote_code": True}
528+
531529
ov_model = OVModelForCausalLM.from_pretrained(model_id, export=True, ov_config=F32_CONFIG, **model_kwargs)
532530
self.assertIsInstance(ov_model.config, PretrainedConfig)
533531
self.assertTrue(ov_model.use_cache)

0 commit comments

Comments
 (0)