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