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.
36 lines
632 B
Python
36 lines
632 B
Python
"""
|
|
Collection of tiling modes
|
|
"""
|
|
|
|
# ruff: noqa: E402
|
|
# pylint: disable=wrong-import-position
|
|
# This is to solve circular imports
|
|
|
|
|
|
class Settings:
|
|
"""
|
|
For representing tiling settings
|
|
"""
|
|
|
|
def __init__(self, mode, rotation):
|
|
self.mode = mode
|
|
self.tiling_fn = modes[mode]
|
|
self.rotation = rotation
|
|
|
|
def __hash__(self):
|
|
return hash((self.mode, self.rotation))
|
|
|
|
|
|
from .hex import hex_tiling
|
|
from .none import none_tiling
|
|
from .rect import rect_tiling
|
|
|
|
modes = {
|
|
"None": none_tiling,
|
|
"Hexagon": hex_tiling,
|
|
"Rectangular": rect_tiling,
|
|
}
|
|
|
|
|
|
__all__ = ["modes", "Settings"]
|