diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index cd59e0365d..92ebc90494 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -648,6 +648,7 @@ class FastModel(FastBaseModel): do_forced_float32 = False model_type_arch = model_types[1] global FORCE_FLOAT32 + print(model_type_arch, FORCE_FLOAT32, dtype) for disable_name in FORCE_FLOAT32: if (disable_name.lower() == model_type_arch.lower() or \ disable_name.lower() in model_name.lower()) and \