From f5e0e98f6df9260a93fb650a0b97c85eb87b0fd3 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Tue, 21 Mar 2023 13:46:36 +0100
Subject: Fixed SNR weighting, re-enabled xformers

---
 environment.yaml | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'environment.yaml')

diff --git a/environment.yaml b/environment.yaml
index 9355f37..db43bd5 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -17,9 +17,11 @@ dependencies:
           - -e git+https://github.com/huggingface/diffusers#egg=diffusers
           - accelerate==0.17.1
           - bitsandbytes==0.37.1
+          - peft==0.2.0
           - python-slugify>=6.1.2
           - safetensors==0.3.0
           - setuptools==65.6.3
           - test-tube>=0.7.5
           - transformers==4.27.1
           - triton==2.0.0
+          - xformers==0.0.17.dev480
-- 
cgit v1.2.3-54-g00ecf