'comfy3d-pt25' add deps for TRELLIS

YanWenKun · Dec 18, 2024 · f82f59e · f82f59e
1 parent f59ad24
commit f82f59e
Show file tree

Hide file tree

Showing 7 changed files with 35 additions and 4 deletions.
diff --git a/comfy3d-pt25/Dockerfile b/comfy3d-pt25/Dockerfile
@@ -30,7 +30,9 @@ COPY builder-scripts/.  /builder-scripts/
 # Deps for ComfyUI & custom nodes
 RUN --mount=type=cache,target=/root/.cache/pip \
     pip install \
-        -r /builder-scripts/pak3.txt
+        -r /builder-scripts/pak3.txt \
+    && pip install \
+        -r /builder-scripts/pak4.txt
 
 RUN --mount=type=cache,target=/root/.cache/pip \
     pip install \
@@ -45,6 +47,8 @@ RUN --mount=type=cache,target=/root/.cache/pip \
         -r /builder-scripts/pak8.txt \
     && pip install \
         -r /builder-scripts/pak9.txt \
+    && pip install \
+        -r /builder-scripts/pakA.txt \
     && pip list
 
 ################################################################################

diff --git a/comfy3d-pt25/builder-scripts/pak4.txt b/comfy3d-pt25/builder-scripts/pak4.txt
@@ -0,0 +1 @@
+spconv-cu124
diff --git a/comfy3d-pt25/builder-scripts/pak5.txt b/comfy3d-pt25/builder-scripts/pak5.txt
@@ -3,10 +3,12 @@ cachetools
 clip-interrogator==0.6.0
 color-matcher
 dill
+easydict
 einops
 fairscale
 GitPython
 gpytoolbox
+igraph
 imageio-ffmpeg
 iopath
 jaxtyping
@@ -29,10 +31,12 @@ PyGithub
 pygltflib
 pyhocon
 PyMCubes
+pymeshfix
 pymeshlab
 python-dotenv
 pytorch-lightning
 pytorch-msssim
+pyvista
 pyyaml
 rembg
 requests

diff --git a/comfy3d-pt25/builder-scripts/pak6.txt b/comfy3d-pt25/builder-scripts/pak6.txt
@@ -4,3 +4,4 @@ git+https://github.com/openai/CLIP.git
 git+https://github.com/WASasquatch/cstr
 git+https://github.com/WASasquatch/ffmpy.git
 git+https://github.com/WASasquatch/img2texture.git
+git+https://github.com/EasternJournalist/utils3d.git#egg=utils3d
diff --git a/comfy3d-pt25/builder-scripts/pak9.txt b/comfy3d-pt25/builder-scripts/pak9.txt
@@ -1 +1,2 @@
-numpy==1.26.4
+--find-links https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124.html
+kaolin
diff --git a/comfy3d-pt25/builder-scripts/pakA.txt b/comfy3d-pt25/builder-scripts/pakA.txt
@@ -0,0 +1 @@
+numpy==1.26.4
diff --git a/comfy3d-pt25/runner-scripts/build-deps.sh b/comfy3d-pt25/runner-scripts/build-deps.sh
@@ -34,15 +34,34 @@ done
 # Compile other deps, using latest
 cd /root
 
-pip install --force-reinstall \
-    "git+https://github.com/ashawkey/diff-gaussian-rasterization.git"
+# pip install --force-reinstall \
+#     "git+https://github.com/ashawkey/diff-gaussian-rasterization.git"
 
 pip install --force-reinstall \
     "git+https://github.com/ashawkey/kiuikit.git"
 
 pip install --force-reinstall \
     "git+https://github.com/NVlabs/nvdiffrast.git"
 
+# For TRELLIS
+# Note: vox2seq is already included in Comfy3D_Pre_Builds.
+
+pip install git+https://github.com/JeffreyXiang/diffoctreerast.git
+
+mkdir -p /tmp/build
+
+git clone --depth=1 https://github.com/autonomousvision/mip-splatting.git \
+    /tmp/build/mip-splatting
+pip uninstall --break-system-packages --yes diff-gaussian-rasterization
+pip install --force-reinstall \
+    /tmp/build/mip-splatting/submodules/diff-gaussian-rasterization/
+
+# (Optional) Compile Flash Attention for Ampere and later GPUs.
+# Limit Ninja jobs to avoid OOM.
+# If have >96GB RAM, just remove MAX_JOBS line.
+export MAX_JOBS=4
+pip install flash-attn --no-build-isolation
+
 # Ensure Numpy1
 pip install numpy==1.26.4