From 38995b97f42a7e391fe5c0e8283e0688ecd8592f Mon Sep 17 00:00:00 2001
From: Luis C <me@lucataco.com>
Date: Tue, 15 Oct 2024 15:46:06 +0000
Subject: [PATCH] Fix lion optimizer

---
 cog.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/cog.yaml b/cog.yaml
index 2674916..8cae855 100644
--- 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"