From 38b5c77635f7a0febb694a45d41489c6c827f564 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sat, 27 Jul 2024 22:54:49 -0700 Subject: [PATCH] Update loader.py --- unsloth/models/loader.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index fc356c3c16..f06515e624 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -98,11 +98,11 @@ def _get_model_name(model_name, load_in_4bit = True): FLOAT_TO_INT_MAPPER = FLOAT_TO_INT_MAPPER, ) print(new_model_name) - if new_model_name is None and \ - model_name.count("/") == 1 and \ - model_name[0].isalnum(): + if new_model_name is None and model_name.count("/") == 1 and model_name[0].isalnum(): # Try checking if a new Unsloth version allows it! NEW_INT_TO_FLOAT_MAPPER, NEW_FLOAT_TO_INT_MAPPER = _get_new_mapper() + print(NEW_INT_TO_FLOAT_MAPPER) + print(NEW_FLOAT_TO_INT_MAPPER) upgraded_model_name = __get_model_name( model_name = model_name, load_in_4bit = load_in_4bit,