63 lines
2.9 KiB
Python
63 lines
2.9 KiB
Python
import logging
|
|
import random
|
|
import re
|
|
from .gpl3 import DarkLoraTagLoader
|
|
from .modules import faces, prompts, files, checkpoints, convert, loras, utils, darkjson
|
|
|
|
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,
|
|
"DarkCheckpointRotator": checkpoints.DarkCheckpointRotator,
|
|
"DarkCheckpointSwitcher": checkpoints.DarkCheckpointSwitcher,
|
|
"DarkCheckpointRotatorIterations": checkpoints.DarkCheckpointRotatorIterations,
|
|
"DarkAnyToString": convert.DarkAnyToString,
|
|
"DarkLoraStackFromString": loras.DarkLoraStackFromString,
|
|
"DarkPopLoraFromStack": loras.DarkPopLoraFromStack,
|
|
"DarkLoRANameFromString": loras.DarkLoRANameFromString,
|
|
"DarkLoRACombineStacks": loras.DarkLoRACombineStacks,
|
|
"DarkJSONLoader": darkjson.loader.DarkJSONLoader,
|
|
"DarkJSONSaver": darkjson.loader.DarkJSONSaver,
|
|
"DarkJSONGetLoRASettings": darkjson.loras.DarkJSONGetLoRASettings,
|
|
"DarkJSONGetLoRACheckpointSettings": darkjson.loras.DarkJSONGetLoRACheckpointSettings,
|
|
"DarkJSONLoRAStackFromJSON": darkjson.loras.DarkLoraStackFromJSON,
|
|
"DarkJSONUpdateLoRA": darkjson.loader.DarkJSONUpdateLoRA,
|
|
"DarkStringToJSON": darkjson.DarkStringToJSON,
|
|
"DarkTextToJSON": darkjson.DarkTextToJSON,
|
|
}
|
|
|
|
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",
|
|
"DarkCheckpointRotator": "Dark Checkpoint Rotator",
|
|
"DarkCheckpointRotatorIterations": "Dark Checkpoint Rotator with Iterations",
|
|
"DarkCheckpointSwitcher": "Dark Checkpoint Switcher",
|
|
"DarkAnyToString": "Dark Any to String",
|
|
"DarkLoraStackFromString": "Dark LoRA Stack from String",
|
|
"DarkPopLoraFromStack": "Dark LoRA Pop LoRA string from LORA_STACK",
|
|
"DarkLoRANameFromString": "Extracts LoRAs from strings and provides various outputs",
|
|
"DarkLoRACombineStacks": "Dark LoRA Combine Stacks",
|
|
"DarkJSONLoader": "Dark JSON Loader",
|
|
"DarkJSONSaver": "Dark JSON Saver",
|
|
"DarkJSONGetLoRASettings": "Dark JSON Get LoRA Settings",
|
|
"DarkJSONGetLoRACheckpointSettings": "Dark JSON Get LoRA Checkpoint Settings",
|
|
"DarkJSONLoRAStackFromJSON": "Dark JSON Get LoRA Stack from JSON",
|
|
"DarkJSONUpdateLoRA": "Dark JSON Update LoRA Data",
|
|
"DarkStringToJSON": "Dark JSON convert string to JSON",
|
|
"DarkTextToJSON": "Dark JSON convert text to JSON",
|
|
}
|
|
|
|
__all__ = [NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS]
|