-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathenvironment.yml
More file actions
executable file
·54 lines (50 loc) · 1.58 KB
/
environment.yml
File metadata and controls
executable file
·54 lines (50 loc) · 1.58 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
name: alphaflow
channels:
- pytorch
- nvidia
dependencies:
- python=3.11
- pip
- pillow
- pytorch=2.6.0
- torchvision
- requests
- pip:
- torch --pre --index-url https://download.pytorch.org/whl/nightly/cu118
- torchvision --pre --index-url https://download.pytorch.org/whl/nightly/cu118
- torchaudio --pre --index-url https://download.pytorch.org/whl/nightly/cu118
- tqdm
- gitpython
- gpustat
- hydra-core==1.3
- scipy==1.10 # For 1.13, FID128 will be computed incorrectly since scipy.linalg.sqrtm will fail for degenerate matrices.
- einops
- av
- ftfy
- regex
- protobuf
- joblib==1.2.0
- wandb==0.16.6
- termcolor
- matplotlib
- click
- torchrl
- torchmetrics==1.3.2
- compressai
- lpips==0.1.4
- timm==1.0.11
- git+https://github.com/openai/CLIP.git
- git+https://github.com/universome/streaming@bf7fca61e3f0c023fb0b3b77187ef017be7f18f0
- beartype
- loguru
- torch_dct
- fvcore # To print FLOPs in model summary
# TODO: for diffusers, we currently need to use our own version with use_reentrant=False in CogVideoX
# Otherwise, gradient checkpointing will not work for it in an FSDP setup.
# - diffusers==0.32.2
- git+https://github.com/universome/diffusers@97715f0dd680594dd6812323df207ad23d8384d6
- git+https://github.com/universome/Cosmos-Tokenizer@ed0cb8542111d3c34c071edc46a5da3339a475dd
# To run some infrastructure scripts
- awscli
# Installing accelerate for the only purpose of diffusers not to through warnings...
- accelerate