diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index cadfed9430..67566fc0b5 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -538,6 +538,7 @@ class FastBaseModel: lora_dropout = lora_dropout, bias = bias, task_type = task_type, + use_rslora = use_rslora, ) model = prepare_model_for_kbit_training( model,