feat: add LoRA training support for Chroma

2026-04-10 15:00:23 +00:00 · 2025-07-20 19:00:09 +09:00
parent c4958b5dca
commit b4e862626a
10 changed files with 158 additions and 260 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -3482,7 +3482,7 @@ def get_sai_model_spec(
    textual_inversion: bool,
    is_stable_diffusion_ckpt: Optional[bool] = None,  # None for TI and LoRA
    sd3: str = None,
-    flux: str = None,
+    flux: str = None, # "dev", "schnell" or "chroma"
 ):
    timestamp = time.time()