diff --git a/torchchat/cli/convert_hf_checkpoint.py b/torchchat/cli/convert_hf_checkpoint.py index c15b3ce82..7001eb1e1 100644 --- a/torchchat/cli/convert_hf_checkpoint.py +++ b/torchchat/cli/convert_hf_checkpoint.py @@ -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: diff --git a/torchchat/cli/download.py b/torchchat/cli/download.py index eb96e99b4..69cfbf21c 100644 --- a/torchchat/cli/download.py +++ b/torchchat/cli/download.py @@ -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