diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index 074e837398..9f2f53aa82 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -79,6 +79,7 @@ def _get_new_mapper(): import requests new_mapper = "https://raw.githubusercontent.com/unslothai/unsloth/main/unsloth/models/mapper.py" with requests.get(new_mapper, timeout = 3) as new_mapper: new_mapper = new_mapper.text + new_mapper = new_mapper[new_mapper.find("__INT_TO_FLOAT_MAPPER"):] new_mapper = new_mapper\ .replace("INT_TO_FLOAT_MAPPER", "NEW_INT_TO_FLOAT_MAPPER")\ .replace("FLOAT_TO_INT_MAPPER", "NEW_FLOAT_TO_INT_MAPPER")