diff --git a/shared.py b/shared.py index 3eb47bc7f..65585c4b0 100644 --- a/shared.py +++ b/shared.py @@ -12,7 +12,7 @@ except: model_kinds = { - "configs": [".yml"], + "configs": [".yml", ".yaml"], "checkpoints": supported_ckpt_extensions, "vae": supported_pt_extensions, "clip": supported_pt_extensions,