diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 8ad5b48883..90b5917b5f 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -1177,6 +1177,7 @@ def unsloth_compile_transformers( return if disable: return + model_types = list(dict().fromkeys(model_types).keys()) for model_type in model_types: _unsloth_compile_transformers( model_type, diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index b9da01c28e..4dfe32dfc7 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -253,10 +253,10 @@ class FastBaseModel: try: vllm_version = f" vLLM: {importlib_version('vllm')}." except: vllm_version = "" - print(model_types) model_type_arch = model_types[0] - if model_type_arch == "siglip" and len(model_types) != 1: - model_type_arch = model_types[1] + if model_type_arch == "siglip": + for model_type_arch in model_types: + if model_type_arch != "siglip": break statistics = \ f"==((====))== Unsloth {__version__}: Fast {model_type_arch.title()} patching. Transformers: {transformers_version}.{vllm_version}\n"\