diff --git a/cli/hf-search.py b/cli/hf-search.py index 53c254d44..f69c215eb 100755 --- a/cli/hf-search.py +++ b/cli/hf-search.py @@ -14,5 +14,5 @@ if __name__ == "__main__": library=['diffusers'], ) res = hf_api.list_models(filter=model_filter, full=True, limit=50, sort="downloads", direction=-1) - models = [{ 'name': m.modelId, 'downloads': m.downloads, 'mtime': m.lastModified, 'url': f'https://huggingface.co/{m.modelId}', 'pipeline': m.pipeline_tag, 'tags': m.tags } for m in res] + models = [{ 'name': m.id, 'downloads': m.downloads, 'mtime': m.lastModified, 'url': f'https://huggingface.co/{m.id}', 'pipeline': m.pipeline_tag, 'tags': m.tags } for m in res] print(models) diff --git a/modules/modelloader.py b/modules/modelloader.py index 02db8b52f..02dbf5c45 100644 --- a/modules/modelloader.py +++ b/modules/modelloader.py @@ -308,7 +308,7 @@ def find_diffuser(name: str): models = list(hf_api.list_models(filter=hf_filter, full=True, limit=20, sort="downloads", direction=-1)) shared.log.debug(f'Searching diffusers models: {name} {len(models) > 0}') if len(models) > 0: - return models[0].modelId + return models[0].id return None diff --git a/modules/ui_models.py b/modules/ui_models.py index a5d3639bf..3ce877177 100644 --- a/modules/ui_models.py +++ b/modules/ui_models.py @@ -362,12 +362,11 @@ def create_ui(): def hf_search(keyword): import huggingface_hub as hf hf_api = hf.HfApi() - model_filter = hf.ModelFilter(model_name=keyword, library=['diffusers']) - models = hf_api.list_models(filter=model_filter, full=True, limit=50, sort="downloads", direction=-1) + models = hf_api.list_models(model_name=keyword, full=True, library="diffusers", limit=50, sort="downloads", direction=-1) data.clear() for model in models: tags = [t for t in model.tags if not t.startswith('diffusers') and not t.startswith('license') and not t.startswith('arxiv') and len(t) > 2] - data.append([model.modelId, model.pipeline_tag, tags, model.downloads, model.lastModified, f'https://huggingface.co/{model.modelId}']) + data.append([model.id, model.pipeline_tag, tags, model.downloads, model.lastModified, f'https://huggingface.co/{model.id}']) return data def hf_select(evt: gr.SelectData, data):