59 lines
2.7 KiB
Python
59 lines
2.7 KiB
Python
# __init__.py
|
|
|
|
|
|
from .FL_KohyaSSInitWorkspace import FL_KohyaSSInitWorkspace
|
|
from .FL_KohyaSSDatasetConfig import FL_KohyaSSDatasetConfig
|
|
from .FL_KohyaSSAdvConfig import FL_KohyaSSAdvConfig
|
|
from .FL_KohyaSSTrain import FL_KohyaSSTrain
|
|
from .FL_LoadImagesFromDirectoryPath import FL_LoadImagesFromDirectoryPath
|
|
from .FL_Kohya_EasyTrain import FL_Kohya_EasyTrain
|
|
from .FL_ImageCaptionSaver import FL_ImageCaptionSaver
|
|
#==============================================================================
|
|
#==============================================================================
|
|
# from .FL_SliderLoraInitWorkspace import FL_SliderLoraInitWorkspace
|
|
# from .FL_SliderLoraDatasetConfig import FL_SliderLoraDatasetConfig
|
|
# from .FL_SliderLoraAdvConfig import FL_SliderLoraAdvConfig
|
|
# from .FL_SliderLoraTrain import FL_SliderLoraTrain
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"FL_KohyaSSInitWorkspace": FL_KohyaSSInitWorkspace,
|
|
"FL_KohyaSSDatasetConfig": FL_KohyaSSDatasetConfig,
|
|
"FL_KohyaSSAdvConfig": FL_KohyaSSAdvConfig,
|
|
"FL_KohyaSSTrain": FL_KohyaSSTrain,
|
|
"FL_LoadImagesFromDirectoryPath": FL_LoadImagesFromDirectoryPath,
|
|
"FL_Kohya_EasyTrain": FL_Kohya_EasyTrain,
|
|
"FL_ImageCaptionSaver": FL_ImageCaptionSaver,
|
|
#==============================================================================
|
|
#==============================================================================
|
|
# "FL_SliderLoraInitWorkspace": FL_SliderLoraInitWorkspace,
|
|
# "FL_SliderLoraDatasetConfig": FL_SliderLoraDatasetConfig,
|
|
# "FL_SliderLoraAdvConfig": FL_SliderLoraAdvConfig,
|
|
# "FL_SliderLoraTrain": FL_SliderLoraTrain
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"FL_KohyaSSInitWorkspace": "FL Kohya Workspace",
|
|
"FL_KohyaSSDatasetConfig": "FL Kohya Dataset Config",
|
|
"FL_KohyaSSAdvConfig": "FL Kohya Adv Config",
|
|
"FL_KohyaSSTrain": "FL Kohya Train",
|
|
"FL_LoadImagesFromDirectoryPath": "FL Kohya Data Loader",
|
|
"FL_Kohya_EasyTrain": "FL Kohya Easy Train",
|
|
"FL_ImageCaptionSaver": "FL Image Caption Saver"
|
|
#==============================================================================
|
|
#==============================================================================
|
|
# "FL_SliderLoraInitWorkspace": "FL Slider LoRA Init Workspace",
|
|
# "FL_SliderLoraDatasetConfig": "FL Slider LoRA Dataset Config",
|
|
# "FL_SliderLoraAdvConfig": "FL Slider LoRA Advanced Config",
|
|
# "FL_SliderLoraTrain": "FL Slider LoRA Train"
|
|
}
|
|
ascii_art = """
|
|
|
|
MACHINE DELUSIONS
|
|
TRAINER LOADED
|
|
|
|
"""
|
|
print(ascii_art)
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"] |