PRM / requirements.txt
JiantaoLin
new
78efc6b
raw
history blame
445 Bytes
torch==2.1.0
torchvision==0.16.0
torchaudio==2.1.0
-f https://download.pytorch.org/whl/cu121
pytorch-lightning==2.1.2
gradio==3.41.2
einops
omegaconf
torchmetrics
webdataset
accelerate==0.30.1
tensorboard
PyMCubes
trimesh
rembg
transformers==4.34.1
diffusers==0.27.2
bitsandbytes
imageio[ffmpeg]
xatlas
plyfile
git+https://github.com/NVlabs/nvdiffrast/
PyGLM==2.7.0
open3d
onnxruntime
xformers==0.0.22.post7
ninja
triton
huggingface_hub==0.23.4