From aab503ac80e2032805854c2ff211cdc2a4b23b3f Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sat, 27 Jul 2024 22:35:24 -0700 Subject: [PATCH] Update loader.py --- unsloth/models/loader.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index 2a7fa75fe4..615b8286fd 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -111,7 +111,8 @@ def _get_model_name(model_name, load_in_4bit = True): if upgraded_model_name is not None: raise NotImplementedError( f"Unsloth: {model_name} is not supported in your current Unsloth version! Please update Unsloth via:\n\n"\ - 'pip install --upgrade --force-reinstall --no-cache-dir "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"' + 'pip uninstall unsloth -y\n'\ + 'pip install --upgrade --no-cache-dir "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"' ) pass pass