Pin specific version of torch/torch-xla

huggingface · Apr 11, 2024 · 8a1cef1 · 8a1cef1
1 parent cba3a11
commit 8a1cef1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -43,8 +43,8 @@ keywords = [
 
 dependencies = [
     "transformers == 4.39.3",
-    "torch-xla[tpu]>=2.2.0",
-    "torch>=2.2.0",
+    "torch == 2.2.0",
+    "torch-xla[tpu] == 2.2.0",
     "loguru == 0.6.0"
 ]