fix use_cookies in get_model_files (#1424)

This commit is contained in:
Xingjun.Wang
2025-07-25 14:24:27 +08:00
committed by GitHub
parent 2b3747db53
commit 49d50b2126

View File

@@ -913,7 +913,13 @@ class HubApi:
Returns:
The query result in bool value
"""
files = self.get_model_files(repo_id, recursive=True, revision=revision)
cookies = ModelScopeConfig.get_cookies()
files = self.get_model_files(
repo_id,
recursive=True,
revision=revision,
use_cookies=False if cookies is None else cookies,
)
files = [file['Path'] for file in files]
return filename in files
@@ -2186,7 +2192,8 @@ class HubApi:
repo_id,
revision=revision or DEFAULT_MODEL_REVISION,
recursive=True,
endpoint=endpoint
endpoint=endpoint,
use_cookies=cookies,
)
file_paths = [f['Path'] for f in files]
elif repo_type == REPO_TYPE_DATASET: