feat(TAESD): support TAESD — Tiny Autoencoder for Stable Diffusion

2024-08-30 20:32:17 +00:00 · 2023-08-17 19:59:31 -07:00
parent 98a4cc20a9
commit 8611ffe32d
2 changed files with 52 additions and 29 deletions
--- a/invokeai/backend/model_management/model_probe.py
+++ b/invokeai/backend/model_management/model_probe.py
@ -51,6 +51,7 @@ class ModelProbe(object):
        "StableDiffusionXLPipeline": ModelType.Main,
        "StableDiffusionXLImg2ImgPipeline": ModelType.Main,
        "AutoencoderKL": ModelType.Vae,
+        "AutoencoderTiny": ModelType.Vae,
        "ControlNetModel": ModelType.ControlNet,
    }