aducsdr commited on
Commit
35218f9
·
verified ·
1 Parent(s): 3b8f136

Create environment.yml

Browse files
Files changed (1) hide show
  1. environment.yml +36 -0
environment.yml ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: seedvr
2
+ channels:
3
+ - pytorch
4
+ - nvidia
5
+ - defaults
6
+ dependencies:
7
+ - python=3.10
8
+ - pytorch=2.1.0
9
+ - torchvision=0.16.0
10
+ - torchaudio=2.1.0
11
+ - pytorch-cuda=12.1
12
+ - pip
13
+ - pip:
14
+ - accelerate==0.24.1
15
+ - av==11.0.0
16
+ - beautifulsoup4==4.12.2
17
+ - einops==0.7.0
18
+ - gradio==4.29.0
19
+ - huggingface_hub
20
+ - iopath==0.1.10
21
+ - omegaconf==2.3.0
22
+ - pandas==2.1.3
23
+ - pytorch-lightning==2.1.0
24
+ - pytorchvideo==0.1.5
25
+ - ftfy==6.1.1
26
+ - regex==2023.10.3
27
+ - timm==0.9.10
28
+ - transformers==4.35.2
29
+ - tqdm==4.66.1
30
+ - webdataset==0.2.70
31
+ - diffusers==0.26.3
32
+ - controlnet_aux==0.0.7
33
+ - opencv-python
34
+ - scikit-image
35
+ - xformers==0.0.22.post7
36
+ - flash_attn==2.5.9.post1 --no-build-isolation