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

Update trainer.py

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

1
finetune/trainer.py
View File

@@ -14,7 +14,6 @@ class CPMTrainer(Trainer):
labels = inputs.pop("labels") labels = inputs.pop("labels")
else: else:
labels = None labels = None
self.model.resampler.pos_embed = self.model.resampler.pos_embed.to(self.model.device)
if not self.args.use_lora: if not self.args.use_lora:
outputs = self.model(data = inputs, use_cache=False) outputs = self.model(data = inputs, use_cache=False)