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

This commit is contained in:
psychedelicious 2024-03-13 16:09:35 +11:00
parent d0800c4888
commit 01207a2fa5

View File

@ -90,8 +90,8 @@ class HuggingFaceMetadataFetch(ModelMetadataFetchBase):
) )
) )
# diffusers models have a `model_index.json` file # diffusers models have a `model_index.json` or `config.json` file
is_diffusers = any(str(f.url).endswith("model_index.json") for f in files) 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 # These URLs will be exposed to the user - I think these are the only file types we fully support
ckpt_urls = ( ckpt_urls = (