Skip to content

Commit 611796f

Browse files
updating filters
1 parent 83ce1df commit 611796f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ads/aqua/common/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1067,7 +1067,7 @@ def get_hf_model_info(repo_id: str) -> ModelInfo:
10671067
@cached(cache=TTLCache(maxsize=1, ttl=timedelta(hours=5), timer=datetime.now))
10681068
def list_hf_models(query:str) -> List[str]:
10691069
try:
1070-
models= HfApi().list_models(model_name=query,task="text-generation")
1070+
models= HfApi().list_models(model_name=query,task="text-generation",sort="downloads",direction=-1,limit=20)
10711071
return [model.id for model in models if model.disabled is None]
10721072
except HfHubHTTPError as err:
10731073
raise format_hf_custom_error_message(err) from err

0 commit comments

Comments
 (0)