ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update loader.py

Browse source
This commit is contained in:
Daniel Han 2024-07-27 22:54:49 -07:00
commit 38b5c77635
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/models/loader.py
View file

@ -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,