Skip to content

Commit

Permalink
remove extra print
Browse files Browse the repository at this point in the history
  • Loading branch information
Gasoonjia committed Sep 24, 2024
1 parent 8ecc2fa commit a70d7b5
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 8 deletions.
2 changes: 0 additions & 2 deletions torchchat/cli/convert_hf_checkpoint.py
Original file line number Diff line number Diff line change
Expand Up @@ -309,8 +309,6 @@ def convert_hf_checkpoint(
model_name: Optional[str] = None,
remove_bin_files: bool = False,
):
print(model_name)
print("***********************")
if "llava" in model_name:
convert_llava_checkpoint(model_dir=model_dir)
else:
Expand Down
6 changes: 0 additions & 6 deletions torchchat/cli/download.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,16 +80,10 @@ def download_and_convert(
# allows recovery in the event that the download or conversion
# fails unexpectedly.
temp_dir = models_dir / "downloads" / model_config.name
# temp_dir = Path("/home/gasoonjia") / "downloads" / model_config.name

if os.path.isdir(temp_dir):
shutil.rmtree(temp_dir)
os.makedirs(temp_dir, exist_ok=True)

print("**************************************************")
print("**************************************************")
print("temp dir: ", temp_dir)

try:
if (
model_config.distribution_channel
Expand Down

0 comments on commit a70d7b5

Please sign in to comment.