38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
__version__ = "0.0.8"
|
|
|
|
import logging
|
|
import random
|
|
import re
|
|
from .gpl3 import DarkLoraTagLoader
|
|
from .modules import faces
|
|
from .modules import prompts
|
|
from .modules import files
|
|
from .modules import checkpoints
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"DarkFolders": files.DarkFolders,
|
|
"DarkCombine": prompts.DarkCombine,
|
|
"DarkPrompt": prompts.DarkPrompt,
|
|
"DarkLoRALoader": DarkLoraTagLoader,
|
|
"DarkFaceIndexShuffle": faces.DarkFaceIndexShuffle,
|
|
"DarkFaceIndexGenerator": faces.DarkFaceIndexGenerator,
|
|
"DarkCheckpointRandomizer": checkpoints.DarkCheckpointRandomizer,
|
|
"DarkCheckpointSwitcher": checkpoints.DarkCheckpointSwitcher,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"DarkFolders": "Dark Folders",
|
|
"DarkCombine": "Dark Combiner",
|
|
"DarkPrompt": "Dark Prompt",
|
|
"DarkLoRALoader": "Dark LoRA Loader",
|
|
"DarkFaceIndexShuffle": "Dark Face Index Shuffle",
|
|
"DarkFaceIndexGenerator": "Dark Face Index Generator",
|
|
"DarkCheckpointRandomizer": "Dark Checkpoint Randomizer",
|
|
"DarkCheckpointSwitcher": "Dark Checkpoint Switcher",
|
|
}
|
|
|
|
__all__ = [NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS]
|