fix(mm): config.json to indicates diffusers model

invoke-ai · Mar 13, 2024 · e97eb7b · e97eb7b
1 parent 74361aa
commit e97eb7b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/invokeai/backend/model_manager/metadata/fetch/huggingface.py b/invokeai/backend/model_manager/metadata/fetch/huggingface.py
@@ -90,8 +90,8 @@ def from_id(self, id: str, variant: Optional[ModelRepoVariant] = None) -> AnyMod
                 )
             )
 
-        # 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 = (