From c927eeb3516b8ecae201441a68956f32247a6d7c Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Fri, 17 Feb 2023 15:56:54 +0100
Subject: Back to xformers

---
 environment.yaml | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

(limited to 'environment.yaml')

diff --git a/environment.yaml b/environment.yaml
index 8010c09..325644f 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -1,17 +1,18 @@
 name: ldd
 channels:
-  - pytorch-nightly
+  - pytorch
   - nvidia
   - xformers/label/dev
   - defaults
 dependencies:
   - cudatoolkit=11.7
+  - libcufile=1.4.0.31
   - matplotlib=3.6.2
   - numpy=1.23.4
   - pip=22.3.1
   - python=3.10.8
-  - pytorch=2.0.0.dev20230216=*cuda*
-  - torchvision=0.15.0.dev20230216
+  - pytorch=1.13.1=*cuda*
+  - torchvision=0.14.1
   - pip:
       - -e .
       - -e git+https://github.com/huggingface/diffusers#egg=diffusers
@@ -22,3 +23,5 @@ dependencies:
       - setuptools==65.6.3
       - test-tube>=0.7.5
       - transformers==4.26.1
+      - triton==2.0.0a2
+      - xformers==0.0.17.dev451
-- 
cgit v1.2.3-54-g00ecf