feat: add AODH Image Saver (Metadata), Lora Selector, Checkpoint Selector, and various node improvements
This commit is contained in:
@@ -2,19 +2,31 @@ from .character_reader import NODE_CLASS_MAPPINGS as CR_CLASS, NODE_DISPLAY_NAME
|
||||
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
|
||||
**LFS_CLASS,
|
||||
**LS_CLASS,
|
||||
**CS_CLASS,
|
||||
**CFS_CLASS,
|
||||
**MS_CLASS
|
||||
}
|
||||
|
||||
NODE_DISPLAY_NAME_MAPPINGS = {
|
||||
**CR_DISPLAY,
|
||||
**RR_DISPLAY,
|
||||
**RE_DISPLAY,
|
||||
**LFS_DISPLAY
|
||||
**LFS_DISPLAY,
|
||||
**LS_DISPLAY,
|
||||
**CS_DISPLAY,
|
||||
**CFS_DISPLAY,
|
||||
**MS_DISPLAY
|
||||
}
|
||||
|
||||
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|
||||
|
||||
Reference in New Issue
Block a user