torch.compile won't work yet, keep code prepared

1 files changed, 31 insertions, 0 deletions

diff --git a/environment_nightly.yaml b/environment_nightly.yaml
new file mode 100644
index 0000000..4c5c798
--- /dev/null
+++ b/environment_nightly.yaml
@@ -0,0 +1,31 @@
+name: ldd
+channels:
+    - pytorch-nightly
+    - nvidia
+    - xformers/label/dev
+    - defaults
+    - conda-forge 
+dependencies:
+    - cuda-nvcc=12.1.105
+    - matplotlib=3.6.2
+    - numpy=1.24.3
+    - pip=22.3.1
+    - python=3.10.8
+    - pytorch=2.1.0.dev20230429=*cuda12.1*
+    - torchvision=0.16.0.dev20230429
+    # - xformers=0.0.19
+    - pip:
+          - -e .
+          - -e git+https://github.com/huggingface/accelerate#egg=accelerate
+          - -e git+https://github.com/huggingface/diffusers#egg=diffusers
+          - -e git+https://github.com/facebookresearch/dadaptation#egg=dadaptation
+          - bitsandbytes==0.38.1
+          - hidet==0.2.3
+          - lion-pytorch==0.0.7
+          - peft==0.2.0
+          - python-slugify>=6.1.2
+          - safetensors==0.3.1
+          - setuptools==65.6.3
+          - test-tube>=0.7.5
+          - timm==0.8.17.dev0
+          - transformers==4.28.1