fix probing for ip_adapter folders

This commit is contained in:
Lincoln Stein 2023-09-23 17:42:11 -04:00 committed by Kent Keirsey
parent e3de996525
commit 86cb53342a

View File

@ -71,7 +71,13 @@ class ModelSearch(ABC):
if any(
[
(path / x).exists()
for x in {"config.json", "model_index.json", "learned_embeds.bin", "pytorch_lora_weights.bin"}
for x in {
"config.json",
"model_index.json",
"learned_embeds.bin",
"pytorch_lora_weights.bin",
"image_encoder.txt",
}
]
):
try: