From dc68caefbdd7cf32eca80d3145a8ef38b1052671 Mon Sep 17 00:00:00 2001 From: qianyu chen <38046403+qyc-98@users.noreply.github.com> Date: Wed, 17 Jul 2024 09:30:28 +0800 Subject: [PATCH] Update trainer.py --- finetune/trainer.py | 1 - 1 file changed, 1 deletion(-) diff --git a/finetune/trainer.py b/finetune/trainer.py index cc45c97..2bf33c4 100644 --- a/finetune/trainer.py +++ b/finetune/trainer.py @@ -14,7 +14,6 @@ class CPMTrainer(Trainer): labels = inputs.pop("labels") else: labels = None - self.model.resampler.pos_embed = self.model.resampler.pos_embed.to(self.model.device) if not self.args.use_lora: outputs = self.model(data = inputs, use_cache=False)