Add support for modules_to_save in FastModel.get_peft_model (#3317)
* patch modules_to_save * Update vision.py --------- Co-authored-by: Daniel Han <danielhanchen@gmail.com>
This commit is contained in:
parent
3117cb7542
commit
026638f5cc
1 changed files with 1 additions and 0 deletions
|
|
@ -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,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue