Fix lion optimizer

replicate · Oct 15, 2024 · 38995b9 · 38995b9
1 parent 3285317
commit 38995b9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cog.yaml b/cog.yaml
@@ -39,6 +39,7 @@ build:
     - "optimum-quanto==0.2.4"
     - "sentencepiece==0.2.0"
     - "peft==0.13.0"
+    - "lion-pytorch==0.2.2"
 
     # llava
     - "deepspeed==0.9.5"