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,