Fix to config

This commit is contained in:
Kent Keirsey
2025-06-23 09:52:47 -04:00
parent b88f4a24d0
commit 61b049ad35

View File

@ -355,7 +355,7 @@ class LoRAOmiConfig(LoRAConfigBase, ModelConfigBase):
base_str, _ = metadata["modelspec.architecture"].split("/")
base_str = base_str.lower()
if "stable-diffusion-v1" in base_str:
if "stable-diffusion-xl-v1-base" in base_str:
base = BaseModelType.StableDiffusionXL
elif "flux" in base_str:
base = BaseModelType.Flux