snomiaoandClaude Opus 4.6
dc2c298ce7
Two nodes: InfiniDepthModelLoader and InfiniDepthEstimate. Supports CUDA, MPS, and CPU via ComfyUI device management. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
25 lines
598 B
Python
25 lines
598 B
Python
"""Post-install script for ComfyUI-InfiniDepth.
|
|
|
|
Installs git-based dependencies that cannot go into requirements.txt.
|
|
"""
|
|
|
|
import subprocess
|
|
import sys
|
|
|
|
|
|
def pip_install(*args):
|
|
subprocess.check_call([sys.executable, "-m", "pip", "install", *args])
|
|
|
|
|
|
try:
|
|
import moge # noqa: F401
|
|
except ImportError:
|
|
print("[InfiniDepth] Installing MoGe …")
|
|
pip_install("git+https://github.com/microsoft/MoGe.git")
|
|
|
|
try:
|
|
import gsplat # noqa: F401
|
|
except ImportError:
|
|
print("[InfiniDepth] Installing gsplat …")
|
|
pip_install("git+https://github.com/nerfstudio-project/gsplat.git")
|