pin hf transformers (#1914)

Lightning-AI · Jan 20, 2025 · 5cd6f04 · 5cd6f04
1 parent a5021be
commit 5cd6f04
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -33,7 +33,7 @@ test = [
     "pytest-rerunfailures>=14.0",
     "pytest-timeout>=2.3.1",
     "pytest-dependency>=0.6.0",
-    "transformers>=4.38.0",  # numerical comparisons
+    "transformers==4.47.1",  # numerical comparisons
     "einops>=0.7.0",
     "protobuf>=4.23.4",
 ]
@@ -50,7 +50,7 @@ all = [
     "tensorboard>=2.14.0",       # litgpt.pretrain
     "torchmetrics>=1.3.1",       # litgpt.pretrain
     "datasets>=2.18.0",          # litgpt.evaluate
-    "transformers>=4.38.0",      # litgpt.evaluate
+    "transformers==4.47.1",      # litgpt.evaluate
     "lm-eval>=0.4.2",            # litgpt.evaluate
     "huggingface_hub[hf_transfer]>=0.21.0",  # download
     "uvloop>=0.2.0 ; sys_platform != 'win32'"  # litdata, only on non-Windows

diff --git a/tests/run_standalone_tests.sh b/tests/run_standalone_tests.sh
@@ -12,7 +12,7 @@ defaults="-m pytest --no-header -v --disable-pytest-warnings --strict-markers --
 echo "Using defaults: ${defaults}"
 
 # find tests marked as `@RunIf(standalone=True)`. done manually instead of with pytest because it is faster
-grep_output=$(grep --recursive --word-regexp . --regexp 'standalone=True' --include '*.py')
+grep_output=$(grep --recursive --word-regexp . --regexp 'standalone=True' --include '*.py' --exclude 'test_thunder*.py')
 
 # file paths, remove duplicates
 files=$(echo "$grep_output" | cut -f1 -d: | sort | uniq)

diff --git a/tests/test_unsloth_executor.py → tests/test_thunder_unsloth_executor.py b/tests/test_unsloth_executor.py → tests/test_thunder_unsloth_executor.py