Use latest optimum to not call deprecated shared_memory API

also WA with stateless to avoid segfault
openvinotoolkit · Mar 20, 2024 · a5fc367 · a5fc367
1 parent 12f4720
commit a5fc367
Show file tree

Hide file tree

Showing 2 changed files with 3 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,7 @@ 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