Minor logging change

This commit is contained in:
oobabooga 2024-03-10 18:20:50 -07:00
parent abcdd0ad5b
commit 15d90d9bd5

View File

@ -294,7 +294,7 @@ def download_model_wrapper(repo_id, specific_file, progress=gr.Progress(), retur
downloader.check_model_files(model, branch, links, sha256, output_folder) downloader.check_model_files(model, branch, links, sha256, output_folder)
progress(1.0) progress(1.0)
else: else:
yield (f"Downloading file{'s' if len(links) > 1 else ''} to `{output_folder}/`") yield (f"Downloading file{'s' if len(links) > 1 else ''} to `{output_folder}`")
downloader.download_model_files(model, branch, links, sha256, output_folder, progress_bar=progress, threads=4, is_llamacpp=is_llamacpp) downloader.download_model_files(model, branch, links, sha256, output_folder, progress_bar=progress, threads=4, is_llamacpp=is_llamacpp)
yield (f"Model successfully saved to `{output_folder}/`.") yield (f"Model successfully saved to `{output_folder}/`.")