diff --git a/unsloth/models/loader_utils.py b/unsloth/models/loader_utils.py index dcb17b41dc..b778b7e95b 100644 --- a/unsloth/models/loader_utils.py +++ b/unsloth/models/loader_utils.py @@ -13,6 +13,12 @@ # limitations under the License. from .mapper import INT_TO_FLOAT_MAPPER, FLOAT_TO_INT_MAPPER, MAP_TO_UNSLOTH_16bit +# https://github.com/huggingface/transformers/pull/26037 allows 4 bit loading! +from packaging.version import Version +from transformers import __version__ as transformers_version +transformers_version = Version(transformers_version) +SUPPORTS_FOURBIT = transformers_version >= Version("4.37") + def __get_model_name( model_name,