Merge remote-tracking branch 'remote/develop' into HEAD

openvinotoolkit · Mar 21, 2024 · 88653aa · 88653aa
2 parents 7288924 + 7974023
commit 88653aa
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/tests/post_training/pipelines/causal_language_model.py b/tests/post_training/pipelines/causal_language_model.py
@@ -25,7 +25,9 @@ class CausalLMHF(PTQTestPipeline):
 
     def prepare_model(self) -> None:
         if self.backend in OV_BACKENDS + [BackendType.FP32]:
-            self.model_hf = OVModelForCausalLM.from_pretrained(self.model_id, export=True, compile=False)
+            self.model_hf = OVModelForCausalLM.from_pretrained(
+                self.model_id, export=True, compile=False, stateful=False
+            )
             self.model = self.model_hf.model
             ov.serialize(self.model, self.fp32_model_dir / "model_fp32.xml")
 

diff --git a/tests/post_training/requirements.txt b/tests/post_training/requirements.txt
@@ -8,8 +8,8 @@ onnxruntime==1.14.1
 pytest==8.0.2
 pytest-cov
 openvino-dev==2024.0.0
-optimum[onnxruntime,openvino]==1.16.0
-optimum-intel @ git+https://github.com/huggingface/optimum-intel@622f585962e5e0ff5f927fd1a96fbc02b60a2e65
+optimum[onnxruntime,openvino]==1.17.1
+optimum-intel==1.15.2
 whowhatbench @ git+https://github.com/andreyanufr/who_what_benchmark@456d3584ce628f6c8605f37cd9a3ab2db1ebf933
 soundfile==0.12.1
 librosa==0.10.0