File tree Expand file tree Collapse file tree 3 files changed +5
-3
lines changed Expand file tree Collapse file tree 3 files changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -771,8 +771,7 @@ def register(
771
771
os .makedirs (local_dir , exist_ok = True )
772
772
# Copy the model from the cache to destination
773
773
snapshot_download (
774
- repo_id = model_name ,
775
- local_dir = local_dir ,
774
+ repo_id = model_name , local_dir = local_dir , local_dir_use_symlinks = False
776
775
)
777
776
# Upload to object storage
778
777
model_artifact_path = upload_folder (
Original file line number Diff line number Diff line change @@ -116,7 +116,7 @@ opctl = [
116
116
" rich" ,
117
117
" fire" ,
118
118
" cachetools" ,
119
- " huggingface_hub"
119
+ " huggingface_hub==0.22.2 "
120
120
]
121
121
optuna = [" optuna==2.9.0" , " oracle_ads[viz]" ]
122
122
spark = [" pyspark>=3.0.0" ]
Original file line number Diff line number Diff line change @@ -577,6 +577,7 @@ def test_import_verified_model(
577
577
mock_snapshot_download .assert_called_with (
578
578
repo_id = hf_model ,
579
579
local_dir = f"{ str (tmpdir )} /{ hf_model } " ,
580
+ local_dir_use_symlinks = False ,
580
581
)
581
582
mock_subprocess .assert_called_with (
582
583
shlex .split (
@@ -779,6 +780,7 @@ def test_import_any_hf_model_custom_container(
779
780
mock_snapshot_download .assert_called_with (
780
781
repo_id = hf_model ,
781
782
local_dir = f"{ str (tmpdir )} /{ hf_model } " ,
783
+ local_dir_use_symlinks = False ,
782
784
)
783
785
mock_subprocess .assert_called_with (
784
786
shlex .split (
@@ -894,6 +896,7 @@ def test_import_any_hf_model_smc_container(
894
896
mock_snapshot_download .assert_called_with (
895
897
repo_id = hf_model ,
896
898
local_dir = f"{ str (tmpdir )} /{ hf_model } " ,
899
+ local_dir_use_symlinks = False ,
897
900
)
898
901
mock_subprocess .assert_called_with (
899
902
shlex .split (
You can’t perform that action at this time.
0 commit comments