Spaces:
Runtime error
Runtime error
Fix model ID sorting
Browse files
utils.py
CHANGED
@@ -87,7 +87,7 @@ def get_compatible_models(task: str, dataset_ids: List[str]) -> List[str]:
|
|
87 |
library=["transformers", "pytorch"],
|
88 |
)
|
89 |
compatible_models.extend(HfApi().list_models(filter=model_filter))
|
90 |
-
return set(
|
91 |
|
92 |
|
93 |
def get_key(col_mapping, val):
|
|
|
87 |
library=["transformers", "pytorch"],
|
88 |
)
|
89 |
compatible_models.extend(HfApi().list_models(filter=model_filter))
|
90 |
+
return sorted(set([model.modelId for model in compatible_models]))
|
91 |
|
92 |
|
93 |
def get_key(col_mapping, val):
|