Skip to content

Commit 785c26f

Browse files
v0xieAcly
authored andcommitted
Avoid replacing found text encoder models with None in GGUF model search
1 parent eb7a717 commit 785c26f

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

ai_diffusion/comfy_client.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,9 @@ async def connect(url=default_url, access_token=""):
134134
clip_models = nodes.options("DualCLIPLoader", "clip_name1")
135135
available_resources.update(_find_text_encoder_models(clip_models))
136136
clip_gguf_models = nodes.options("DualCLIPLoaderGGUF", "clip_name1")
137-
available_resources.update(_find_text_encoder_models(clip_gguf_models))
137+
for k, v in _find_text_encoder_models(clip_gguf_models).items():
138+
if available_resources.get(k) is None and v is not None:
139+
available_resources[k] = v
138140

139141
vae_models = nodes.options("VAELoader", "vae_name")
140142
available_resources.update(_find_vae_models(vae_models))

0 commit comments

Comments
 (0)