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: Returns:
The query result in bool value 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] files = [file['Path'] for file in files]
return filename in files return filename in files
@@ -2186,7 +2192,8 @@ class HubApi:
repo_id, repo_id,
revision=revision or DEFAULT_MODEL_REVISION, revision=revision or DEFAULT_MODEL_REVISION,
recursive=True, recursive=True,
endpoint=endpoint endpoint=endpoint,
use_cookies=cookies,
) )
file_paths = [f['Path'] for f in files] file_paths = [f['Path'] for f in files]
elif repo_type == REPO_TYPE_DATASET: elif repo_type == REPO_TYPE_DATASET: