diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 7a07a8d0fd..1451ed92cd 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -646,6 +646,7 @@ class FastBaseModel: "lora_dropout" : lora_dropout, "bias" : bias, "task_type" : task_type, + "modules_to_save" : modules_to_save, "use_rslora" : use_rslora, "init_lora_weights" : init_lora_weights, "loftq_config" : loftq_config,