Jordan Thompson
59f2ffc26e
Refactor model class into stand-alone general class. Add Checkpoint Loader
14 lines
461 B
Python
14 lines
461 B
Python
from .civitai_lora_loader import CivitAI_LORA_Loader
|
|
from .civitai_checkpoint_loader import CivitAI_Checkpoint_Loader
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"CivitAI_Lora_Loader": CivitAI_LORA_Loader,
|
|
"CivitAI_Checkpoint_Loader": CivitAI_Checkpoint_Loader
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"CivitAI_Lora_Loader": "CivitAI Lora Loader",
|
|
"CivitAI_Checkpoint_Loader": "CivitAI Checkpoint Loader"
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] |