Update optimum version (#893)

* Update optimum version * trigger * fix
huggingface · Sep 10, 2024 · 2b621cf · 2b621cf
1 parent 24edf41
commit 2b621cf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@
 INSTALL_REQUIRE = [
     "torch>=1.11",
     "transformers>=4.36,<4.45",
-    "optimum@git+https://github.com/huggingface/optimum.git",
+    "optimum~=1.22",
     "datasets>=1.4.0",
     "sentencepiece",
     "setuptools",

diff --git a/tests/openvino/test_training.py b/tests/openvino/test_training.py
@@ -805,7 +805,7 @@ def test_training(self, _, desc: OVTrainerTestDescriptor):
         self.run_ovtrainer_training_checks(desc)
 
     def prepare_model_and_dataset(self, desc: OVTrainerTestDescriptor):
-        self.dataset = load_dataset("anton-l/superb_dummy", "ks")
+        self.dataset = load_dataset("anton-l/superb_dummy", "ks", trust_remote_code=True)
         self.num_labels = len(self.dataset["test"].features["label"].names)
 
         self.feature_extractor = AutoFeatureExtractor.from_pretrained(desc.model_id)