From 01207a2fa50fc056848a2b1a90aad0bba9d4d8b5 Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Wed, 13 Mar 2024 16:09:35 +1100 Subject: [PATCH] fix(mm): `config.json` to indicates diffusers model --- invokeai/backend/model_manager/metadata/fetch/huggingface.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/invokeai/backend/model_manager/metadata/fetch/huggingface.py b/invokeai/backend/model_manager/metadata/fetch/huggingface.py index 0afcd96029..4e3625fdbe 100644 --- a/invokeai/backend/model_manager/metadata/fetch/huggingface.py +++ b/invokeai/backend/model_manager/metadata/fetch/huggingface.py @@ -90,8 +90,8 @@ class HuggingFaceMetadataFetch(ModelMetadataFetchBase): ) ) - # diffusers models have a `model_index.json` file - is_diffusers = any(str(f.url).endswith("model_index.json") for f in files) + # diffusers models have a `model_index.json` or `config.json` file + is_diffusers = any(str(f.url).endswith(("model_index.json", "config.json")) for f in files) # These URLs will be exposed to the user - I think these are the only file types we fully support ckpt_urls = (