28 lines
816 B
Python
28 lines
816 B
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 via kohya sd-scripts.
|
|
"""
|
|
|
|
from .realtime_lora_trainer import RealtimeLoraTrainer, ApplyTrainedLora
|
|
from .sdxl_lora_trainer import SDXLLoraTrainer
|
|
|
|
# Web directory for JavaScript extensions
|
|
WEB_DIRECTORY = "./web/js"
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"RealtimeLoraTrainer": RealtimeLoraTrainer,
|
|
"ApplyTrainedLora": ApplyTrainedLora,
|
|
"SDXLLoraTrainer": SDXLLoraTrainer,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"RealtimeLoraTrainer": "Realtime LoRA Trainer",
|
|
"ApplyTrainedLora": "Apply Trained LoRA",
|
|
"SDXLLoraTrainer": "Realtime LoRA Trainer (SDXL - sd-scripts)",
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'WEB_DIRECTORY']
|