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

Merge pull request #257 from KabakaWilliam/main

Browse Source 
imported deepseed twice. Removed unnecesary extra import
This commit is contained in:
Hongji Zhu
2024-06-17 20:32:58 +08:00
committed by GitHub
parent ceb8c53c91 7017aa1754
commit 472686b3c6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
finetune/trainer.py
View File

@@ -5,9 +5,9 @@ from transformers import Trainer
from transformers.trainer_pt_utils import nested_detach
from transformers.utils import is_sagemaker_mp_enabled
from transformers.trainer import *
import deepspeed
from transformers.integrations import is_deepspeed_zero3_enabled
class CPMTrainer(Trainer):
def compute_loss(self, model, inputs, return_outputs=False):
if "labels" in inputs: