From ce2baa36a97bca63ec9126b9289586afb2cda249 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Sun, 8 Oct 2023 19:49:03 -0400 Subject: [PATCH] port support for AutoencoderTiny models --- invokeai/backend/model_manager/probe.py | 1 + 1 file changed, 1 insertion(+) diff --git a/invokeai/backend/model_manager/probe.py b/invokeai/backend/model_manager/probe.py index 82dc7c722b..8be0233def 100644 --- a/invokeai/backend/model_manager/probe.py +++ b/invokeai/backend/model_manager/probe.py @@ -95,6 +95,7 @@ class ModelProbe(ModelProbeBase): "StableDiffusionXLPipeline": ModelType.Main, "StableDiffusionXLImg2ImgPipeline": ModelType.Main, "AutoencoderKL": ModelType.Vae, + "AutoencoderTiny": ModelType.Vae, "ControlNetModel": ModelType.ControlNet, "CLIPVisionModelWithProjection": ModelType.CLIPVision, "T2IAdapter": ModelType.T2IAdapter,