Kelly Yang
c96d6e30b4
Stage workflows/tests/assets/cube.obj into ComfyUI input during install, run validation plus execution_light on CPU, and exercise Load3D/Preview3D with a real mesh.
85 lines
2.6 KiB
Python
85 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
|
"""Install TRI-ML RecGen inference under vendor/recgen (not custom_nodes/)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
RECGEN_REPO = "https://github.com/TRI-ML/RecGen.git"
|
|
|
|
|
|
def _node_root() -> Path:
|
|
return Path(__file__).resolve().parent
|
|
|
|
|
|
def _vendor_recgen() -> Path:
|
|
return _node_root() / "vendor" / "recgen"
|
|
|
|
|
|
def _comfyui_root() -> Path:
|
|
"""ComfyUI root when install.py runs from custom_nodes/ComfyUI-Recgen/."""
|
|
return _node_root().parent.parent
|
|
|
|
|
|
def _stage_test_3d_assets() -> None:
|
|
"""Copy bundled test OBJ into ComfyUI input/3d for Load3D workflows."""
|
|
cube_src = _node_root() / "workflows" / "tests" / "assets" / "cube.obj"
|
|
if not cube_src.is_file():
|
|
return
|
|
input_3d = _comfyui_root() / "input" / "3d"
|
|
input_3d.mkdir(parents=True, exist_ok=True)
|
|
dest = input_3d / "cube.obj"
|
|
shutil.copy2(cube_src, dest)
|
|
print(f"[ComfyUI-Recgen] Staged test 3D model at input/3d/cube.obj")
|
|
|
|
|
|
def _pip_install(*args: str) -> None:
|
|
"""Install into the active interpreter (ComfyUI venv in comfy-test CI)."""
|
|
attempts = [
|
|
[sys.executable, "-m", "pip", "install", *args],
|
|
["uv", "pip", "install", "--python", sys.executable, *args],
|
|
]
|
|
errors: list[str] = []
|
|
for cmd in attempts:
|
|
try:
|
|
subprocess.check_call(cmd)
|
|
return
|
|
except FileNotFoundError:
|
|
errors.append(f"command not found: {cmd[0]}")
|
|
except subprocess.CalledProcessError as exc:
|
|
errors.append(f"{' '.join(cmd)} failed (exit {exc.returncode})")
|
|
raise RuntimeError(
|
|
"Could not install dependencies via pip or uv pip:\n "
|
|
+ "\n ".join(errors)
|
|
)
|
|
|
|
|
|
def ensure_recgen() -> Path:
|
|
node_root = _node_root()
|
|
recgen = _vendor_recgen()
|
|
marker = recgen / "recgen_inference"
|
|
if not marker.is_dir():
|
|
recgen.parent.mkdir(parents=True, exist_ok=True)
|
|
print(f"[ComfyUI-Recgen] Cloning RecGen into {recgen} ...")
|
|
subprocess.check_call(
|
|
["git", "clone", "--depth", "1", RECGEN_REPO, str(recgen)],
|
|
)
|
|
# Node deps (trimesh, etc.) must be in ComfyUI's venv, not only the host runner.
|
|
print(f"[ComfyUI-Recgen] Installing ComfyUI-Recgen package deps into {sys.executable} ...")
|
|
_pip_install("-e", str(node_root))
|
|
print(f"[ComfyUI-Recgen] Installing recgen_inference from {recgen} ...")
|
|
_pip_install("-e", str(recgen))
|
|
_stage_test_3d_assets()
|
|
return recgen
|
|
|
|
|
|
def main() -> None:
|
|
ensure_recgen()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|