-
Notifications
You must be signed in to change notification settings - Fork 8
/
requirements.txt
51 lines (46 loc) · 802 Bytes
/
requirements.txt
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
--extra-index-url https://download.pytorch.org/whl/cu117
torch==1.13.1
torchvision==0.14.1
pytorch-lightning==2.0.0
omegaconf==2.3.0
jaxtyping
typeguard
diffusers[torch]==0.19.3 # <20.0
transformers==4.28.1
accelerate==0.24.1
opencv-python
tensorboard
matplotlib
imageio>=2.28.0
imageio[ffmpeg]
libigl
xatlas
trimesh[easy]
networkx
pysdf
PyMCubes
wandb
gradio
git+https://github.com/ashawkey/envlight.git
torchmetrics==1.2.0
# deepfloyd
xformers==0.0.16 # 0.0.12
bitsandbytes==0.38.1
sentencepiece
safetensors
huggingface_hub
# for zero123
einops
kornia
taming-transformers-rom1504
git+https://github.com/openai/CLIP.git
#controlnet
controlnet_aux
# candidates
ftfy==6.1.1
rembg
triton==2.1.0
lpips==0.1.4
torch-efficient-distloss==0.1.3
nerfacc==0.5.2
git+https://github.com/NVlabs/nvdiffrast.git