From 8abbd633d8ee7500058b2f1f69a6d6611b5a4450 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Fri, 17 Mar 2023 15:18:20 +0100
Subject: Test: https://arxiv.org/pdf/2303.09556.pdf

---
 environment.yaml | 46 ++++++++++++++++++++++------------------------
 1 file changed, 22 insertions(+), 24 deletions(-)

(limited to 'environment.yaml')

diff --git a/environment.yaml b/environment.yaml
index 018a9ab..9355f37 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -1,27 +1,25 @@
 name: ldd
 channels:
-  - pytorch
-  - nvidia
-  - xformers/label/dev
-  - defaults
+    - 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=1.13.1=*cuda*
-  - torchvision=0.14.1
-  - xformers=0.0.17.dev466
-  - pip:
-      - -e .
-      - -e git+https://github.com/huggingface/diffusers#egg=diffusers
-      - accelerate==0.16.0
-      - bitsandbytes==0.37.0
-      - 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.0
+    - matplotlib=3.6.2
+    - numpy=1.23.4
+    - pip=22.3.1
+    - python=3.10.8
+    - pytorch=2.0.0=*cuda11.8*
+    - torchvision=0.15.0
+    # - xformers=0.0.17.dev476
+    - pip:
+          - -e .
+          - -e git+https://github.com/huggingface/diffusers#egg=diffusers
+          - accelerate==0.17.1
+          - bitsandbytes==0.37.1
+          - 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
-- 
cgit v1.2.3-54-g00ecf