diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 41da912600..ab53811f49 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -__version__ = "2025.3.16" +__version__ = "2025.3.17" __all__ = [ "SUPPORTS_BFLOAT16", @@ -1175,11 +1175,9 @@ def unsloth_compile_transformers( "so turning off some optimizations!" ) return - print(disable) if disable: return model_types = list(dict().fromkeys(model_types).keys()) - print(model_types) for model_type in model_types: _unsloth_compile_transformers( model_type, @@ -1212,7 +1210,6 @@ def unsloth_compile_transformers( # Redo patches which override compiler for temporary_patch in TEMPORARY_PATCHES: temporary_patch() - print(os.environ["UNSLOTH_FORCE_FLOAT32"]) return model_types pass diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index 20b96eb2c5..670e082580 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -650,7 +650,6 @@ class FastModel(FastBaseModel): if model_type_arch != "siglip": break global FORCE_FLOAT32 for disable_name in FORCE_FLOAT32: - print(model_types, disable_name.lower(), model_type_arch.lower(), model_name.lower(), dtype, SUPPORTS_BFLOAT16) if (disable_name.lower() == model_type_arch.lower() or \ disable_name.lower() in model_name.lower()) and \ ((dtype == torch.float16) or not SUPPORTS_BFLOAT16): @@ -658,7 +657,6 @@ class FastModel(FastBaseModel): dtype = torch.bfloat16 # Change to bfloat16 loading break pass - print(model_type_arch, FORCE_FLOAT32, dtype, os.environ["UNSLOTH_FORCE_FLOAT32"]) # Patch gradient checkpointing if use_gradient_checkpointing == "unsloth": patch_unsloth_smart_gradient_checkpointing(dtype = dtype)