33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
from .character_reader import NODE_CLASS_MAPPINGS as CR_CLASS, NODE_DISPLAY_NAME_MAPPINGS as CR_DISPLAY
|
|
from .resolution_reader import NODE_CLASS_MAPPINGS as RR_CLASS, NODE_DISPLAY_NAME_MAPPINGS as RR_DISPLAY
|
|
from .reenforcer import NODE_CLASS_MAPPINGS as RE_CLASS, NODE_DISPLAY_NAME_MAPPINGS as RE_DISPLAY
|
|
from .lora_from_string import NODE_CLASS_MAPPINGS as LFS_CLASS, NODE_DISPLAY_NAME_MAPPINGS as LFS_DISPLAY
|
|
from .lora_selector import NODE_CLASS_MAPPINGS as LS_CLASS, NODE_DISPLAY_NAME_MAPPINGS as LS_DISPLAY
|
|
from .checkpoint_selector import NODE_CLASS_MAPPINGS as CS_CLASS, NODE_DISPLAY_NAME_MAPPINGS as CS_DISPLAY
|
|
from .checkpoint_from_string import NODE_CLASS_MAPPINGS as CFS_CLASS, NODE_DISPLAY_NAME_MAPPINGS as CFS_DISPLAY
|
|
from .metadata_saver import NODE_CLASS_MAPPINGS as MS_CLASS, NODE_DISPLAY_NAME_MAPPINGS as MS_DISPLAY
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
**CR_CLASS,
|
|
**RR_CLASS,
|
|
**RE_CLASS,
|
|
**LFS_CLASS,
|
|
**LS_CLASS,
|
|
**CS_CLASS,
|
|
**CFS_CLASS,
|
|
**MS_CLASS
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
**CR_DISPLAY,
|
|
**RR_DISPLAY,
|
|
**RE_DISPLAY,
|
|
**LFS_DISPLAY,
|
|
**LS_DISPLAY,
|
|
**CS_DISPLAY,
|
|
**CFS_DISPLAY,
|
|
**MS_DISPLAY
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|