Skip to content

v0.1.7 — standalone load_pipeline fix

Choose a tag to compare

@dreamrec dreamrec released this 14 May 16:18

Fix: Standalone install actually works now

v0.1.6 shipped to the Comfy Registry but failed at first run with a misleading
Repository Not Found for url: https://huggingface.co/ckpts/... 404. v0.1.7
fixes the real root cause (three layers deep) and adds standalone-mode install support.

Root cause analysis

  1. comfy-aimdo's mmgp lib (shipped in ComfyUI Desktop's .venv)
    monkey-patches safetensors.torch.load_file with a memory-mapped reader.
    Its dtype map at mmgp/safetensors2.py:33 omits complex dtypes; one
    Pixal3D decoder checkpoint contains a single C64 (complex64) tensor →
    KeyError: 'C64' blows up the load.
  2. Upstream Pixal3D's pipelines/base.py:43-46 has a bare except Exception
    that silently swallows that KeyError and retries with the relative path.
  3. The retry path in models/__init__.py:60 misparses relative paths like
    "ckpts/ss_flow_..." as Hugging Face repo IDs and emits the bogus 404 we see.

Fixes (nodes/pixal3d_stages.py)

  • snapshot_download(model_path) resolves to an absolute local snapshot dir
    before from_pretrained, so the loader's os.path.exists check (relative
    to CWD) always succeeds.
  • Pre-extend mmgp.safetensors2._map_to_dtype with C64/C128 so the mmap
    reader handles complex tensors.
  • Wrap from_pretrained in torch.inference_mode(False) as belt-and-suspenders
    against load_state_dict version-counter errors on freshly-constructed
    nn.Module parameters under ComfyUI's outer inference_mode.

Fixes (install.py)

  • Standalone-mode fallback: if no pozzettiandrea/ComfyUI-TRELLIS2
    sibling exists, install deps into sys.executable instead of hard-failing.
    This matches how Comfy Registry / Manager invoke install.py with the
    .venv's python — fresh Registry installs now work out of the box.
  • Worker python search prefers C:\ce\_env_* (the real comfy-env worker
    pool) over any in-plugin stub Scripts/python.exe.
  • install_requirements splits requirements.txt into VCS / hashed / plain
    buckets so pip's --require-hashes mode doesn't reject the VCS MoGe entry.
  • Error message: visualbrunopozzettiandrea (correct upstream fork).

Verified

End-to-end on ComfyUI Desktop + RTX 5090: 56.5s warm run with 8/8/8 steps +
1024 texture + low_vram=True → 8.6 MB GLB output.

Full Changelog: v0.1.6...v0.1.7