From 242383371b9649081a5fd060277eed3ef054745f Mon Sep 17 00:00:00 2001 From: Daniel Bevenius Date: Fri, 28 Jun 2024 09:02:54 +0200 Subject: [PATCH] convert-hf : print output file name when completed This commit adds the output file name to the log message when the conversion is completed. The motivation for this change is that when `--outfile` option is not specified it migth not be obvious where the output file is written. With this change the output of running the script will be something like the following: ```console INFO:hf-to-gguf:Model successfully exported to models/gemma-2-9b-it.gguf. ``` Signed-off-by: Daniel Bevenius --- convert-hf-to-gguf.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index 5bcc849db..3c974ae09 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -3128,11 +3128,11 @@ def main() -> None: if args.vocab_only: logger.info("Exporting model vocab...") model_instance.write_vocab() - logger.info("Model vocab successfully exported.") + logger.info(f"Model vocab successfully exported to {fname_out}.") else: logger.info("Exporting model...") model_instance.write() - logger.info("Model successfully exported.") + logger.info(f"Model successfully exported to {fname_out}.") if __name__ == '__main__':