Skip to content

Commit 8c75754

Browse files
authored
remove obsolete param for using symlinks (#1314)
1 parent 95ebcb8 commit 8c75754

File tree

1 file changed

+11
-4
lines changed

1 file changed

+11
-4
lines changed

torchchat/cli/download.py

+11-4
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,10 @@
1010
from pathlib import Path
1111
from typing import Optional
1212

13-
from torchchat.cli.convert_hf_checkpoint import convert_hf_checkpoint, convert_hf_checkpoint_to_tune
13+
from torchchat.cli.convert_hf_checkpoint import (
14+
convert_hf_checkpoint,
15+
convert_hf_checkpoint_to_tune,
16+
)
1417
from torchchat.model_config.model_config import (
1518
load_model_configs,
1619
ModelConfig,
@@ -57,7 +60,6 @@ def _download_hf_snapshot(
5760
snapshot_download(
5861
model_config.distribution_path,
5962
local_dir=artifact_dir,
60-
local_dir_use_symlinks=False,
6163
token=hf_token,
6264
ignore_patterns=ignore_patterns,
6365
)
@@ -77,9 +79,14 @@ def _download_hf_snapshot(
7779
raise e
7880

7981
# Convert the Multimodal Llama model to the torchtune format.
80-
if model_config.name in {"meta-llama/Llama-3.2-11B-Vision-Instruct", "meta-llama/Llama-3.2-11B-Vision"}:
82+
if model_config.name in {
83+
"meta-llama/Llama-3.2-11B-Vision-Instruct",
84+
"meta-llama/Llama-3.2-11B-Vision",
85+
}:
8186
print(f"Converting {model_config.name} to torchtune format...", file=sys.stderr)
82-
convert_hf_checkpoint_to_tune( model_dir=artifact_dir, model_name=model_config.name)
87+
convert_hf_checkpoint_to_tune(
88+
model_dir=artifact_dir, model_name=model_config.name
89+
)
8390

8491
else:
8592
# Convert the model to the torchchat format.

0 commit comments

Comments
 (0)