associate sdxl config with sdxl VAEs

This commit is contained in:
Lincoln Stein
2024-06-19 23:43:56 -04:00
parent b03073d888
commit 349239e336

View File

@@ -312,6 +312,8 @@ class ModelProbe(object):
config_file = (
"stable-diffusion/v1-inference.yaml"
if base_type is BaseModelType.StableDiffusion1
else "stable-diffusion/sd_xl_base.yaml"
if base_type is BaseModelType.StableDiffusionXL
else "stable-diffusion/v2-inference.yaml"
)
else: