From 66f3673d9d28b68c08fcf885a442c6243a3632bf Mon Sep 17 00:00:00 2001 From: Dmitry Date: Fri, 7 Mar 2025 15:44:03 +0100 Subject: [PATCH 1/2] Conftest: get_device_name() without device occupation (#1826) --- conftest.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/conftest.py b/conftest.py index 23c6df78b5..1ce796e44f 100644 --- 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" From 93ade6a1756b0aa9f3b1104dd977cd955429c36a Mon Sep 17 00:00:00 2001 From: ZhengHongming888 Date: Fri, 7 Mar 2025 06:45:26 -0800 Subject: [PATCH 2/2] Fix dataset_version for ST example requirement.txt (#1809) --- examples/sentence-transformers-training/nli/requirements.txt | 2 +- .../sentence-transformers-training/paraphrases/requirements.txt | 2 +- examples/sentence-transformers-training/sts/requirements.txt | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/sentence-transformers-training/nli/requirements.txt b/examples/sentence-transformers-training/nli/requirements.txt index 680dc8a2bb..1b97e4c3d7 100644 --- 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 index aee11b288a..b776a8dd19 100644 --- 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 index 680dc8a2bb..1b97e4c3d7 100644 --- 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