We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 103620a commit ebc108bCopy full SHA for ebc108b
optimum/intel/openvino/modeling_base.py
@@ -448,6 +448,9 @@ def half(self):
448
self.request = None
449
return self
450
451
+ def eval(self):
452
+ return self
453
+
454
def forward(self, *args, **kwargs):
455
raise NotImplementedError
456
tests/openvino/test_modeling.py
@@ -115,6 +115,7 @@ def test_load_from_hub_and_save_model(self):
115
tokenizer = AutoTokenizer.from_pretrained(self.OV_MODEL_ID)
116
tokens = tokenizer("This is a sample input", return_tensors="pt")
117
loaded_model = OVModelForSequenceClassification.from_pretrained(self.OV_MODEL_ID)
118
+ loaded_model.eval()
119
self.assertIsInstance(loaded_model.config, PretrainedConfig)
120
loaded_model_outputs = loaded_model(**tokens)
121
0 commit comments