Pin optimum package (axolotl-ai-cloud#838)

mkeoliya · Nov 10, 2023 · 64af6a3 · 64af6a3
1 parent d95b605
commit 64af6a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -17,7 +17,7 @@ sentencepiece
 wandb
 einops
 xformers>=0.0.22
-optimum
+optimum==1.13.2
 hf_transfer
 colorama
 numba