From 9d8bb08270d7c2cc1bf7b81ee63dd5f73b5b3027 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 26 Nov 2024 00:47:27 -0800 Subject: [PATCH] prints --- unsloth/models/_utils.py | 2 ++ unsloth/models/loader.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index ac9c9a98ca..eed61b30c0 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -1142,12 +1142,14 @@ def unsloth_compile_transformers( pass if disable: return + print(1) model_types = get_transformers_model_type( model_name = model_name, token = token, revision = revision, trust_remote_code = trust_remote_code, ) + print(model_types) for model_type in model_types: _unsloth_compile_transformers( model_type, diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index b33c1e19db..9d9ac31b6e 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -358,7 +358,9 @@ class FastVisionModel(FastBaseVisionModel): patch_unsloth_smart_gradient_checkpointing() old_model_name = model_name + print(model_name) model_name = get_model_name(model_name, load_in_4bit) + print(model_name) with contextlib.redirect_stdout(open(os.devnull, "w")): patch_loss_functions(torch_compile = False)