Update save.py

This commit is contained in:
Daniel Han-Chen 2024-03-11 02:25:10 +11:00
commit 133097d995

View file

@ -577,6 +577,21 @@ def install_llama_cpp_old(version = -10):
latest = releases[-1]
version = releases[version].split(" ")[0]
# Check if the llama.cpp exists
if os.path.exists("llama.cpp"):
print(
"**[WARNING]** You have a llama.cpp old directory which is broken.\n"\
"Unsloth will DELETE the broken directory and install a new one.\n"\
"Press CTRL + C / cancel this if this is wrong. We shall wait 10 seconds.\n"
)
import time
for i in range(10):
print(f"**[WARNING]** Deleting llama.cpp directory... {10-i} seconds left.")
time.sleep(1)
import shutil
shutil.rmtree("llama.cpp")
pass
# Clone a specific commit
commands = [
"git clone https://github.com/ggerganov/llama.cpp",