avtc
01a324bb11
Extends tiling beyond UNet (Conv2d) models to support DiT architectures using toroidal attention patches and latent content wrapping for seamless infinite tiling. Adds rectangular tiling mode, WanVAE 5D tensor handling, and an AdvancedTilingRay node for distributed Raylight workers.
34 lines
882 B
Python
34 lines
882 B
Python
"""
|
|
ComfyUI Node entry point
|
|
"""
|
|
|
|
# pylint: disable=invalid-name
|
|
|
|
from .advanced_tiling import (
|
|
AdvancedTilingSettings,
|
|
AdvancedTiling,
|
|
AdvancedTilingVAEDecode,
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"AdvancedTilingSettings": AdvancedTilingSettings,
|
|
"AdvancedTiling": AdvancedTiling,
|
|
"AdvancedTilingVAEDecode": AdvancedTilingVAEDecode,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"AdvancedTilingSettings": "Advanced Tiling Settings",
|
|
"AdvancedTiling": "Advanced Tiling",
|
|
"AdvancedTilingVAEDecode": "Advanced Tiling VAE Decode",
|
|
}
|
|
|
|
try:
|
|
from .ray_tiling import AdvancedTilingRay, HAS_RAYLIGHT
|
|
if HAS_RAYLIGHT:
|
|
NODE_CLASS_MAPPINGS["AdvancedTilingRay"] = AdvancedTilingRay
|
|
NODE_DISPLAY_NAME_MAPPINGS["AdvancedTilingRay"] = "Advanced Tiling (Raylight)"
|
|
except ImportError:
|
|
pass
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|