Update loader.py

This commit is contained in:
Daniel Han 2024-07-27 23:04:23 -07:00
commit ea0a49448d

View file

@ -98,7 +98,6 @@ def _get_model_name(model_name, load_in_4bit = True):
INT_TO_FLOAT_MAPPER = INT_TO_FLOAT_MAPPER,
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():
# Try checking if a new Unsloth version allows it!
NEW_INT_TO_FLOAT_MAPPER, NEW_FLOAT_TO_INT_MAPPER = _get_new_mapper()