diff --git a/scripts/unsloth_install.py b/scripts/unsloth_install.py index 1ac26d349..a6570b4e9 100644 --- a/scripts/unsloth_install.py +++ b/scripts/unsloth_install.py @@ -32,5 +32,5 @@ raise RuntimeError(f"Torch = {v} too new!") x = x.format(cuda.replace(".", ""), "-ampere" if is_ampere else "") print( - f'pip install unsloth-zoo && pip install --no-deps "unsloth[{x}] @ git+https://github.com/unslothai/unsloth.git"' + f'pip install unsloth-zoo==2024.11.7 && pip install --no-deps "unsloth[{x}]==2024.11.9"' )