Skip to content

Commit ebc108b

Browse files
committed
Add eval method to OVModels
1 parent 103620a commit ebc108b

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

optimum/intel/openvino/modeling_base.py

+3
Original file line numberDiff line numberDiff line change
@@ -448,6 +448,9 @@ def half(self):
448448
self.request = None
449449
return self
450450

451+
def eval(self):
452+
return self
453+
451454
def forward(self, *args, **kwargs):
452455
raise NotImplementedError
453456

tests/openvino/test_modeling.py

+1
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,7 @@ def test_load_from_hub_and_save_model(self):
115115
tokenizer = AutoTokenizer.from_pretrained(self.OV_MODEL_ID)
116116
tokens = tokenizer("This is a sample input", return_tensors="pt")
117117
loaded_model = OVModelForSequenceClassification.from_pretrained(self.OV_MODEL_ID)
118+
loaded_model.eval()
118119
self.assertIsInstance(loaded_model.config, PretrainedConfig)
119120
loaded_model_outputs = loaded_model(**tokens)
120121

0 commit comments

Comments
 (0)