diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index 6eb8ed7863..6d388900de 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -490,6 +490,7 @@ class FastVisionModel(FastBaseVisionModel): device_map = device_map, trust_remote_code = trust_remote_code, revision = revision if not is_peft else None, + model_types = model_types, *args, **kwargs, )