43 lines
1.8 KiB
Python
43 lines
1.8 KiB
Python
"""
|
|
ComfyUI Realtime LoRA Trainer
|
|
|
|
Trains LoRAs on-the-fly from images during generation.
|
|
Supports Z-Image, FLUX, Wan models via AI-Toolkit.
|
|
Also supports SDXL and SD 1.5 via kohya sd-scripts.
|
|
"""
|
|
|
|
from .realtime_lora_trainer import RealtimeLoraTrainer, ApplyTrainedLora
|
|
from .sdxl_lora_trainer import SDXLLoraTrainer
|
|
from .sd15_lora_trainer import SD15LoraTrainer
|
|
from .musubi_zimage_lora_trainer import MusubiZImageLoraTrainer
|
|
from .musubi_qwen_image_lora_trainer import MusubiQwenImageLoraTrainer
|
|
from .musubi_qwen_image_edit_lora_trainer import MusubiQwenImageEditLoraTrainer
|
|
from .musubi_wan_lora_trainer import MusubiWanLoraTrainer
|
|
|
|
# Web directory for JavaScript extensions
|
|
WEB_DIRECTORY = "./web/js"
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"RealtimeLoraTrainer": RealtimeLoraTrainer,
|
|
"ApplyTrainedLora": ApplyTrainedLora,
|
|
"SDXLLoraTrainer": SDXLLoraTrainer,
|
|
"SD15LoraTrainer": SD15LoraTrainer,
|
|
"MusubiZImageLoraTrainer": MusubiZImageLoraTrainer,
|
|
"MusubiQwenImageLoraTrainer": MusubiQwenImageLoraTrainer,
|
|
"MusubiQwenImageEditLoraTrainer": MusubiQwenImageEditLoraTrainer,
|
|
"MusubiWanLoraTrainer": MusubiWanLoraTrainer,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"RealtimeLoraTrainer": "Realtime LoRA Trainer",
|
|
"ApplyTrainedLora": "Apply Trained LoRA",
|
|
"SDXLLoraTrainer": "Realtime LoRA Trainer (SDXL - sd-scripts)",
|
|
"SD15LoraTrainer": "Realtime LoRA Trainer (SD 1.5 - sd-scripts)",
|
|
"MusubiZImageLoraTrainer": "Realtime LoRA Trainer (Z-Image - Musubi Tuner)",
|
|
"MusubiQwenImageLoraTrainer": "Realtime LoRA Trainer (Qwen Image - Musubi Tuner)",
|
|
"MusubiQwenImageEditLoraTrainer": "Realtime LoRA Trainer (Qwen Image Edit - Musubi Tuner)",
|
|
"MusubiWanLoraTrainer": "Realtime LoRA Trainer (Wan 2.2 - Musubi Tuner)",
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'WEB_DIRECTORY']
|