Skip to content

Commit f3e8402

Browse files
if-aiif-ai
authored
Merge pull request #51 from jhj0517/fix/add-dependencies
Add some dependencies
2 parents 094bc3a + a4545ce commit f3e8402

File tree

4 files changed

+75
-7
lines changed

4 files changed

+75
-7
lines changed

‎pyproject.toml‎

Lines changed: 45 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,51 @@ name = "comfyui-if_trellis"
33
description = "ComfyUI IF Trellis creates a 3d mesh from a single view or multi angle pictures"
44
version = "0.2.1"
55
license = { file = "LICENSE.txt" }
6-
dependencies = ["pillow", "imageio", "imageio-ffmpeg", "tqdm", "easydict","opencv-python-headless","scipy","ninja","plyfile","rembg","onnxruntime-gpu","trimesh","xatlas","pyvista","pymeshfix","igraph","sageattention","git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8","xformers","spconv-cu120"]
6+
dependencies = [
7+
"pillow",
8+
"imageio",
9+
"imageio-ffmpeg",
10+
"tqdm",
11+
"easydict",
12+
"opencv-python-headless",
13+
"scipy",
14+
"ninja",
15+
"plyfile",
16+
"rembg",
17+
"onnxruntime-gpu",
18+
"trimesh",
19+
"xatlas",
20+
"pyvista",
21+
"pymeshfix",
22+
"igraph",
23+
"sageattention",
24+
"git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8",
25+
"xformers",
26+
"spconv-cu124",
27+
"diffrp-nvdiffrast",
28+
29+
# triton for linux
30+
'triton; sys_platform == "linux"',
31+
# triton for windows
32+
'https://github.com/woct0rdho/triton-windows/releases/download/v3.1.0-windows.post8/triton-3.1.0-cp312-cp312-win_amd64.whl; sys_platform == "win64" and python_version=="3.12"',
33+
'https://github.com/woct0rdho/triton-windows/releases/download/v3.1.0-windows.post8/triton-3.1.0-cp311-cp311-win_amd64.whl; sys_platform == "win64" and python_version=="3.11"',
34+
'https://github.com/woct0rdho/triton-windows/releases/download/v3.1.0-windows.post8/triton-3.1.0-cp310-cp310-win_amd64.whl; sys_platform == "win64" and python_version=="3.10"',
35+
'https://github.com/woct0rdho/triton-windows/releases/download/v3.1.0-windows.post8/triton-3.1.0-cp38-cp38-win_amd64.whl; sys_platform == "win64" and python_version=="3.8"',
36+
37+
# koalin for linux
38+
'https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124/kaolin-0.17.0-cp312-cp312-linux_x86_64.whl; sys_platform == "linux" and python_version=="3.12"',
39+
'https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124/kaolin-0.17.0-cp311-cp311-linux_x86_64.whl; sys_platform == "linux" and python_version=="3.11"',
40+
'https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124/kaolin-0.17.0-cp310-cp310-linux_x86_64.whl; sys_platform == "linux" and python_version=="3.10"',
41+
'https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124/kaolin-0.17.0-cp39-cp39-linux_x86_64.whl; sys_platform == "linux" and python_version=="3.9"',
42+
# koalin for windows
43+
'https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124/kaolin-0.17.0-cp312-cp312-win_amd64.whl; sys_platform == "win64" and python_version=="3.12"',
44+
'https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124/kaolin-0.17.0-cp311-cp311-win_amd64.whl; sys_platform == "win64" and python_version=="3.11"',
45+
'https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124/kaolin-0.17.0-cp312-cp310-win_amd64.whl; sys_platform == "win64" and python_version=="3.10"',
46+
'https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124/kaolin-0.17.0-cp312-cp39-win_amd64.whl; sys_platform == "win64" and python_version=="3.9"',
47+
48+
# `diff_gaussian_rasterization` from source code
49+
"git+https://github.com/graphdeco-inria/diff-gaussian-rasterization.git",
50+
]
751

852
[project.urls]
953
Repository = "https://github.com/if-ai/ComfyUI-IF_Trellis.git"

‎requirements.txt‎

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,10 +23,17 @@ pymeshfix
2323
igraph
2424
git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
2525
xformers
26-
#kaolin>=0.17.0
27-
#spconv-cu124>=2.3.6
26+
triton
27+
spconv-cu124>=2.3.6
2828
#transformers>=4.46.3
2929
sageattention
3030
#git+https://github.com/Dao-AILab/flash-attention
31-
#https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl?download=true
32-
#https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl?download=true
31+
git+https://github.com/graphdeco-inria/diff-gaussian-rasterization.git
32+
diffrp-nvdiffrast
33+
34+
# koalin ( CUDA 12.4 specific builds from : https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124.html )
35+
# If you're not using CUDA 12.4, read official documentation at : https://kaolin.readthedocs.io/en/latest/notes/installation.html
36+
https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124/kaolin-0.17.0-cp312-cp312-linux_x86_64.whl;python_version=="3.12"
37+
https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124/kaolin-0.17.0-cp311-cp311-linux_x86_64.whl;python_version=="3.11"
38+
https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124/kaolin-0.17.0-cp310-cp310-linux_x86_64.whl;python_version=="3.10"
39+
https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124/kaolin-0.17.0-cp39-cp39-linux_x86_64.whl;python_version=="3.9"

‎trellis/renderers/gaussian_render.py‎

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -75,8 +75,6 @@ def render(viewpoint_camera, pc : Gaussian, pipe, bg_color : torch.Tensor, scali
7575
image_width=int(viewpoint_camera.image_width),
7676
tanfovx=tanfovx,
7777
tanfovy=tanfovy,
78-
kernel_size=kernel_size,
79-
subpixel_offset=subpixel_offset,
8078
bg=bg_color,
8179
scale_modifier=scaling_modifier,
8280
viewmatrix=viewpoint_camera.world_view_transform,

‎win_requirements.txt‎

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,3 +13,22 @@ pyvista
1313
pymeshfix
1414
igraph
1515
spconv-cu124
16+
17+
# triton ( Windows specific builds from : https://github.com/woct0rdho/triton-windows )
18+
https://github.com/woct0rdho/triton-windows/releases/download/v3.1.0-windows.post8/triton-3.1.0-cp312-cp312-win_amd64.whl;python_version=="3.12"
19+
https://github.com/woct0rdho/triton-windows/releases/download/v3.1.0-windows.post8/triton-3.1.0-cp311-cp311-win_amd64.whl;python_version=="3.11"
20+
https://github.com/woct0rdho/triton-windows/releases/download/v3.1.0-windows.post8/triton-3.1.0-cp310-cp310-win_amd64.whl;python_version=="3.10"
21+
https://github.com/woct0rdho/triton-windows/releases/download/v3.1.0-windows.post8/triton-3.1.0-cp38-cp38-win_amd64.whl;python_version=="3.8"
22+
23+
# koalin ( CUDA 12.4 specific builds from : https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124.html )
24+
# If you're not using CUDA 12.4, read official documentation at : https://kaolin.readthedocs.io/en/latest/notes/installation.html
25+
https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124/kaolin-0.17.0-cp312-cp312-win_amd64.whl;python_version=="3.12"
26+
https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124/kaolin-0.17.0-cp311-cp311-win_amd64.whl;python_version=="3.11"
27+
https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124/kaolin-0.17.0-cp310-cp310-win_amd64.whl;python_version=="3.10"
28+
https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124/kaolin-0.17.0-cp39-cp39-win_amd64.whl;python_version=="3.9"
29+
30+
# diff_gaussian_rasterization
31+
git+https://github.com/graphdeco-inria/diff-gaussian-rasterization.git
32+
33+
# diffrp-nvdiffrast
34+
diffrp-nvdiffrast

0 commit comments

Comments
 (0)