From e487abd94ab82f1acd8a25aba06be6241884c841 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Mon, 5 Feb 2024 19:07:00 +1100 Subject: [PATCH] Update save.py --- unsloth/save.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/unsloth/save.py b/unsloth/save.py index 30827962ec..efffbf5979 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -902,6 +902,7 @@ def unsloth_save_pretrained_gguf( makefile = install_llama_cpp_make_non_blocking() new_save_directory = unsloth_save_model(**arguments) python_install.wait() + makefile.wait() else: try: new_save_directory = unsloth_save_model(**arguments) @@ -914,6 +915,7 @@ def unsloth_save_pretrained_gguf( makefile = install_llama_cpp_make_non_blocking() new_save_directory = unsloth_save_model(**arguments) python_install.wait() + makefile.wait() pass pass