update

2025-12-06 20:44:33 +08:00 · 2024-07-22 16:01:27 +00:00
1 changed files with 2 additions and 3 deletions
--- a/utils/fetch_torch_cuda_pipeline_test_matrix.py
+++ b/utils/fetch_torch_cuda_pipeline_test_matrix.py
@@ -4,7 +4,7 @@ import os
 from collections import defaultdict
 from pathlib import Path

-from huggingface_hub import HfApi, ModelFilter
+from huggingface_hub import HfApi

 import diffusers

@@ -27,7 +27,6 @@ PIPELINE_USAGE_CUTOFF = int(os.getenv("PIPELINE_USAGE_CUTOFF", 50000))

 logger = logging.getLogger(__name__)
 api = HfApi()
-filter = ModelFilter(library="diffusers")


 def filter_pipelines(usage_dict, usage_cutoff=10000):
@@ -46,7 +45,7 @@ def filter_pipelines(usage_dict, usage_cutoff=10000):


 def fetch_pipeline_objects():
-    models = api.list_models(filter=filter)
+    models = api.list_models(library="diffusers")
    downloads = defaultdict(int)

    for model in models: