gokaygokay commited on
Commit
d89b0d6
·
verified ·
1 Parent(s): c0bc540

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +17 -20
requirements.txt CHANGED
@@ -1,26 +1,23 @@
1
- accelerate==1.0.1
2
- bitsandbytes==0.44.1
3
- deepspeed
4
- diffusers==0.19.3
5
  einops
6
- huggingface-hub
7
- imageio[ffmpeg]
8
- invisible_watermark
9
  omegaconf
10
- peft
11
- plyfile
 
 
 
12
  PyMCubes
13
- pytorch-lightning==2.1.2
14
  rembg
15
- sentencepiece
16
- tensorboard
17
- torch==2.1.0 --index-url https://download.pytorch.org/whl/cu121
18
- torchvision==0.16.0
19
- torchaudio==2.1.0
20
- torchmetrics
21
  transformers==4.34.1
22
- trimesh
23
- webdataset
 
24
  xatlas
25
- xformers
26
- git+https://github.com/NVlabs/nvdiffrast/
 
 
 
1
+ torch==2.1.0
2
+ torchvision==0.16.0
3
+ torchaudio==2.1.0
4
+ pytorch-lightning==2.1.2
5
  einops
 
 
 
6
  omegaconf
7
+ deepspeed
8
+ torchmetrics
9
+ webdataset
10
+ accelerate
11
+ tensorboard
12
  PyMCubes
13
+ trimesh
14
  rembg
 
 
 
 
 
 
15
  transformers==4.34.1
16
+ diffusers==0.19.3
17
+ bitsandbytes
18
+ imageio[ffmpeg]
19
  xatlas
20
+ plyfile
21
+ xformers==0.0.22.post7
22
+ git+https://github.com/NVlabs/nvdiffrast/
23
+ huggingface-hub