update dependencies for galore

axolotl-ai-cloud · Mar 19, 2024 · 9437b48 · 9437b48
1 parent dc45927
commit 9437b48
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 --extra-index-url https://huggingface.github.io/autogptq-index/whl/cu118/
 packaging==23.2
 peft==0.9.0
-transformers==4.38.2
+transformers @ git+https://github.com/huggingface/transformers.git@f6261d7d81edd036fc53bfede65fe91f01a661aa
 tokenizers==0.15.0
 bitsandbytes>=0.43.0
 accelerate==0.26.1

diff --git a/setup.py b/setup.py
@@ -89,6 +89,8 @@ def parse_requirements():
         "lion-pytorch": [
             "lion-pytorch==0.1.2",
         ],
-        "galore": ["galore_torch @ git+https://github.com/jiaweizzhao/GaLore"],
+        "galore": [
+            "galore_torch",
+        ],
     },
 )