Yuwei Guo 2 年之前
父节点
当前提交
887ec4494c
共有 1 个文件被更改,包括 7 次插入7 次删除
  1. 7 7
      environment.yaml

+ 7 - 7
environment.yaml

@@ -1,18 +1,18 @@
 name: animatediff
 channels:
   - pytorch
-  - xformers
+  - nvidia
 dependencies:
   - python=3.10
-  - pytorch==1.12.1
-  - torchvision==0.13.1
-  - torchaudio==0.12.1
-  - cudatoolkit=11.3
-  - xformers
+  - pytorch=1.13.1
+  - torchvision=0.14.1
+  - torchaudio=0.13.1
+  - pytorch-cuda=11.7
   - pip
   - pip:
-    - diffusers[torch]==0.11.1
+    - diffusers==0.11.1
     - transformers==4.25.1
+    - xformers==0.0.16
     - imageio==2.27.0
     - gdown
     - einops