deepgeek/MiniCPM-o
2
0
Fork 0
mirror of https://github.com/OpenBMB/MiniCPM-V.git synced 2026-02-05 18:29:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update finetune.py

Browse Source
This commit is contained in:
qianyu chen
2024-07-17 09:30:35 +08:00
committed by GitHub
parent dc68caefbd
commit cb8d4fc5a4
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
finetune/finetune.py
View File

@@ -260,6 +260,7 @@ def train():
lora_dropout=lora_args.lora_dropout, lora_dropout=lora_args.lora_dropout,
bias=lora_args.lora_bias, bias=lora_args.lora_bias,
layers_to_transform=lora_args.lora_layers_to_transform, layers_to_transform=lora_args.lora_layers_to_transform,
modules_to_save=modules_to_save,
) )
if not hasattr(model, 'get_input_embeddings'): if not hasattr(model, 'get_input_embeddings'):
def get_input_embeddings(self): def get_input_embeddings(self):