forked from xxlong0/Wonder3D
-
Notifications
You must be signed in to change notification settings - Fork 0
/
requirements.txt
44 lines (43 loc) · 920 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
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.4.1
torchvision
diffusers[torch]==0.30.0
transformers>=4.25.1
bitsandbytes==0.35.4
decord==0.6.0
pytorch-lightning<2
omegaconf==2.2.3
nerfacc==0.3.3
trimesh==3.9.8
pyhocon==0.3.57
icecream==2.1.0
# PyMCubes==0.1.2
accelerate
modelcards
einops
ftfy
piq
matplotlib
opencv-python
imageio
imageio-ffmpeg
scipy
pyransac3d
torch_efficient_distloss
tensorboard
rembg
segment_anything
gradio==3.50.2
mosaicml-streaming
onnxruntime_gpu==1.17.0
# onnxruntime_gpu @ https://pkgs.dev.azure.com/onnxruntime/2a773b67-e88b-4c7f-9fc0-87d31fea8ef2/_packaging/7fa31e42-5da1-4e84-a664-f2b4129c7d45/pypi/download/onnxruntime-gpu/1.17/onnxruntime_gpu-1.17.0-cp310-cp310-manylinux_2_28_x86_64.whl
pyrender
jaxtyping
pymeshlab
cholespy
torch-scatter
git+https://github.com/facebookresearch/pytorch3d.git@stable
git+https://github.com/NVlabs/nvdiffrast.git
open3d
fire
comet_ml