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,