From 4f724ca8015771c55ab9f382ebec5fd8b3309eb2 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sat, 24 Jun 2023 20:13:03 +0200 Subject: Added Prodigy optimizer --- environment.yaml | 1 + 1 file changed, 1 insertion(+) (limited to 'environment.yaml') diff --git a/environment.yaml b/environment.yaml index 2c81a90..4a73688 100644 --- a/environment.yaml +++ b/environment.yaml @@ -27,6 +27,7 @@ dependencies: - bitsandbytes==0.39.1 - lion-pytorch==0.0.7 - peft==0.3.0 + - prodigyopt==1.0 - python-slugify>=6.1.2 - safetensors==0.3.1 - setuptools==65.6.3 -- cgit v1.2.3-54-g00ecf