Reuses the validated TRELLIS HiCache model-level forecast patch, adapted to TRELLIS.2's five flow DiTs (sparse_structure + shape_slat 512/1024 + tex_slat 512/1024). GPU-validated on TRELLIS.2-4B/RTX 5090 (512 pipeline, interval=2): 1.9x at Chamfer 0.0107 (near-lossless). 8 standalone unit tests. Targets the visualbruno/ComfyUI-Trellis2 TRELLIS2PIPELINE type.
8 lines
310 B
Python
8 lines
310 B
Python
"""ComfyUI-TRELLIS-HiCache: training-free HiCache acceleration for TRELLIS."""
|
|
try:
|
|
from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
|
|
except ImportError:
|
|
from nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|