add case for clip embed models in probe

This commit is contained in:
maryhipp 2024-08-21 14:26:17 -04:00 committed by Brandon
parent 5fed573a29
commit 5f567f41f4

View File

@ -107,6 +107,7 @@ class ModelProbe(object):
"ControlNetModel": ModelType.ControlNet,
"CLIPVisionModelWithProjection": ModelType.CLIPVision,
"T2IAdapter": ModelType.T2IAdapter,
"CLIPModel": ModelType.CLIPEmbed,
}
@classmethod