1 files changed, 1 insertions, 2 deletions

diff --git a/environment.yaml b/environment.yaml
index 57624a3..1e6ac60 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -13,16 +13,15 @@ dependencies:
   - python=3.10.8
   - pytorch=1.13.1=*cuda*
   - torchvision=0.14.1
+  - xformers=0.0.17.dev461
   - pip:
       - -e .
       - -e git+https://github.com/huggingface/diffusers#egg=diffusers
       - accelerate==0.16.0
       - bitsandbytes==0.37.0
-      - lion-pytorch==0.0.6
       - python-slugify>=6.1.2
       - safetensors==0.2.8
       - setuptools==65.6.3
       - test-tube>=0.7.5
       - transformers==4.26.1
       - triton==2.0.0a2
-      - xformers==0.0.17.dev451