Merge pull request #50 from replicate/lion-fix

Fix lion optimizer
replicate · Oct 15, 2024 · 0bf7bb1 · 0bf7bb1
2 parents 3285317 + 38995b9
commit 0bf7bb1
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"