From bba36b7d5eee35ba18e3e66e41cedaa13afddb90 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 26 Nov 2024 03:09:32 -0800 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index c906ded5a4..6118ab926b 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -1142,14 +1142,14 @@ def unsloth_compile_transformers( pass if disable: return - print(model_name) + 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,