Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-05-18 05:05:00 +02:00
Code Issues Packages Projects Releases Wiki Activity

make local_file_only configurable (#1632)

Browse Source 
Signed-off-by: wangli <wangli858794774@gmail.com>
This commit is contained in:
Li Wang
2026-03-09 13:03:22 +08:00
committed by GitHub
parent 918a808dc2
commit dbbcbf631f
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modelscope/utils/hf_util/patcher.py
View File

@@ -180,9 +180,11 @@ def _patch_pretrained_class(all_imported_modules, wrap=False):
revision = 'master'
if file_filter is not None:
allow_file_pattern = file_filter
local_files_only = kwargs.pop('local_files_only', False)
model_dir = snapshot_download(
pretrained_model_name_or_path,
revision=revision,
local_files_only=local_files_only,
ignore_file_pattern=ignore_file_pattern,
allow_file_pattern=allow_file_pattern)
if subfolder: