mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-21 02:29:23 +01:00
fix use_cookies in get_model_files (#1424)
This commit is contained in:
@@ -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:
|
||||||
|
|||||||
Reference in New Issue
Block a user