Merge pull request #243 from Wauplin/patch-1

Update download.py (remove deprecated use of `ModelFilter`)
gitmylo · Aug 16, 2024 · 5c55c38 · 5c55c38
2 parents 9eb161d + c98c4c3
commit 5c55c38
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/webui/modules/download.py b/webui/modules/download.py
@@ -38,7 +38,7 @@ def fill_models(model_type: str):
     if model_type == 'rvc':
         return get_rvc_models()
     return [model.id for model in
-            huggingface_hub.list_models(filter=huggingface_hub.ModelFilter(task=model_type), sort='downloads')]
+            huggingface_hub.list_models(task=model_type, sort='downloads')]
 
 
 def get_file_name(repo_id: str):