File size: 962 Bytes
6f63bf4 45bcac4 032e1ee 45bcac4 6f63bf4 45bcac4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | from .svdquant_capture import (
NODE_CLASS_MAPPINGS as _CAPTURE_NODES,
NODE_DISPLAY_NAME_MAPPINGS as _CAPTURE_NAMES,
)
from .svdquant_diag import (
NODE_CLASS_MAPPINGS as _DIAG_NODES,
NODE_DISPLAY_NAME_MAPPINGS as _DIAG_NAMES,
)
from .svdquant_lora import (
NODE_CLASS_MAPPINGS as _LORA_NODES,
NODE_DISPLAY_NAME_MAPPINGS as _LORA_NAMES,
)
from .svdquant_quantize import (
NODE_CLASS_MAPPINGS as _QUANT_NODES,
NODE_DISPLAY_NAME_MAPPINGS as _QUANT_NAMES,
)
from .svdquant_w4a4 import (
NODE_CLASS_MAPPINGS as _W4A4_NODES,
NODE_DISPLAY_NAME_MAPPINGS as _W4A4_NAMES,
)
NODE_CLASS_MAPPINGS = {**_LORA_NODES, **_W4A4_NODES, **_DIAG_NODES, **_QUANT_NODES,
**_CAPTURE_NODES}
NODE_DISPLAY_NAME_MAPPINGS = {**_LORA_NAMES, **_W4A4_NAMES, **_DIAG_NAMES, **_QUANT_NAMES,
**_CAPTURE_NAMES}
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|