mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
fix probing for ip_adapter folders (#4669)
## What type of PR is this? (check all applicable) - [X] Bug Fix - [ ] Optimizatio ## Have you discussed this change with the InvokeAI team? - [ ] Yes - [X] Np ## Have you updated all relevant documentation? - [ ] Yes - [X] No ## Description ip_adapter models live in a folder containing the file `image_encoder.txt` and a safetensors file. The load-time probe for new models was detecting the files contained within the folder rather than the folder itself, and so models.yaml was not getting correctly updated. This fixes the issue. ## Added/updated tests? - [ ] Yes - [ ] No : _please replace this line with details on why tests have not been included_ ## [optional] Are there any post deployment tasks we need to perform?
This commit is contained in:
commit
35ae9f6e71
@ -71,7 +71,13 @@ class ModelSearch(ABC):
|
|||||||
if any(
|
if any(
|
||||||
[
|
[
|
||||||
(path / x).exists()
|
(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:
|
try:
|
||||||
|
Loading…
Reference in New Issue
Block a user