Fix model probing of CLIP-G model with CLIPTextModelWithProjection class type.

2026-04-23 03:00:31 -04:00 · 2024-10-23 14:01:30 +00:00
parent b3b930a6f5
commit a7d8db8c15
1 changed files with 1 additions and 0 deletions
--- a/invokeai/backend/model_manager/probe.py
+++ b/invokeai/backend/model_manager/probe.py
@@ -121,6 +121,7 @@ class ModelProbe(object):
        "T2IAdapter": ModelType.T2IAdapter,
        "CLIPModel": ModelType.CLIPEmbed,
        "CLIPTextModel": ModelType.CLIPEmbed,
+        "CLIPTextModelWithProjection": ModelType.CLIPEmbed,
        "T5EncoderModel": ModelType.T5Encoder,
        "FluxControlNetModel": ModelType.ControlNet,
    }