duplicate-repo / requirements.txt
Adarsh7700's picture
Update requirements.txt
73558ac verified
raw
history blame contribute delete
328 Bytes
torch==2.1.0
torchvision==0.16.0
torchaudio==2.1.0
pytorch-lightning==2.1.2
einops
huggingface-hub==0.25.1
omegaconf
deepspeed
torchmetrics
webdataset
accelerate
tensorboard
PyMCubes
trimesh
rembg
transformers
diffusers
bitsandbytes
imageio[ffmpeg]
xatlas
plyfile
xformers==0.0.22.post7
git+https://github.com/NVlabs/nvdiffrast/