Drop-in LoRA loader with optional per-layer SVD energy trimming (kohya resize math) and hot-layer taming to reduce artifacts from stacked LoRAs and turbo/distilled models. Passes through DoRA/tucker/unknown keys and hands the cleaned state dict to ComfyUI's standard load_lora_for_models.
7 lines
151 B
Python
7 lines
151 B
Python
from .lora_loader import (
|
|
NODE_CLASS_MAPPINGS,
|
|
NODE_DISPLAY_NAME_MAPPINGS,
|
|
)
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|