PRM / requirements.txt
LTT's picture
Update requirements.txt
c07b3f8 verified
raw
history blame
482 Bytes
torch==2.2.0
torchvision==0.17.0
torchaudio==2.2.0
-f https://download.pytorch.org/whl/cu121
pytorch-lightning==2.1.2
gradio==4.32.0
einops
omegaconf
torchmetrics
webdataset
accelerate==0.30.1
tensorboard
PyMCubes
trimesh
rembg
transformers==4.36
tokenizers
diffusers==0.25
bitsandbytes
imageio[ffmpeg]
xatlas
plyfile
# nvdiffrast @ git+https://github.com/NVlabs/nvdiffrast
# nvdiffrast==0.3.1
PyGLM==2.7.0
open3d
onnxruntime
xformers==0.0.24
ninja
triton
huggingface-hub==0.25.2