20231088/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[Bugfix] Modify modelscope api usage in transformer_utils (#13807)

Browse Source
This commit is contained in:
Shanshan Shen 2025-02-25 16:36:07 +08:00 committed by GitHub
parent aab392774b
commit 2d87d7d1ac
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
vllm/transformers_utils/utils.py
View File

@ -29,9 +29,8 @@ def modelscope_list_repo_files(
) -> List[str]: ) -> List[str]:
"""List files in a modelscope repo.""" """List files in a modelscope repo."""
from modelscope.hub.api import HubApi from modelscope.hub.api import HubApi
from modelscope.utils.hf_util import _try_login
_try_login(token)
api = HubApi() api = HubApi()
api.login(token)
# same as huggingface_hub.list_repo_files # same as huggingface_hub.list_repo_files
files = [ files = [
file['Path'] for file in api.get_model_files( file['Path'] for file in api.get_model_files(