merge with upstream

truefoundry · Dec 22, 2024 · e1e7e52 · e1e7e52
2 parents 8a893f0 + 5950240
commit e1e7e52
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
 --extra-index-url https://download.pytorch.org/whl/cu121
 -r base-requirements.txt
 unsloth[cu121-ampere-torch250] @ git+https://github.com/unslothai/unsloth.git@9dc399a6b6625ee40835c5eab361426d3c5d4abb
-axolotl[deepspeed,flash-attn,mamba-ssm,optimizers,lion-pytorch,galore] @ git+https://github.com/truefoundry/axolotl@5878daa3beec58bf4f4d21a6abd6dba3c40e74f4
+axolotl[deepspeed,flash-attn,optimizers,lion-pytorch,galore] @ git+https://github.com/truefoundry/axolotl@5878daa3beec58bf4f4d21a6abd6dba3c40e74f4