Merge branch 'main' into synapse_1_20

huggingface · Mar 7, 2025 · 73dd3ed · 73dd3ed
2 parents 81f33ed + 93ade6a
commit 73dd3ed
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 9 deletions.
diff --git a/conftest.py b/conftest.py
@@ -110,14 +110,11 @@ def pytest_sessionstart(session):
         device = "gaudi2" if os.environ["GAUDI2_CI"] == "1" else "gaudi1"
     # Try to automatically detect it
     else:
-        import habana_frameworks.torch.hpu as torch_hpu
+        from optimum.habana.utils import get_device_name
 
-        name = torch_hpu.get_device_name().strip()
-        if not name:
-            raise RuntimeError("Expected a Gaudi device but did not detect one.")
-        device = name.split()[-1].lower()
+        device = get_device_name()
 
-    # torch_hpu.get_device_name() returns GAUDI for G1
+    # optimum.habana.utils.get_device_name() returns `gaudi` for G1
     if "gaudi" == device:
         # use "gaudi1" since this is used in tests, baselines, etc.
         device = "gaudi1"

diff --git a/examples/sentence-transformers-training/nli/requirements.txt b/examples/sentence-transformers-training/nli/requirements.txt
@@ -1,2 +1,2 @@
-datasets
+datasets <= 2.19.2
 peft
diff --git a/examples/sentence-transformers-training/paraphrases/requirements.txt b/examples/sentence-transformers-training/paraphrases/requirements.txt
@@ -1 +1 @@
-datasets
+datasets <= 2.19.2
diff --git a/examples/sentence-transformers-training/sts/requirements.txt b/examples/sentence-transformers-training/sts/requirements.txt
@@ -1,2 +1,2 @@
-datasets
+datasets <= 2.19.2
 peft