mirror of
https://github.com/OpenBMB/MiniCPM-V.git
synced 2026-02-04 09:49:20 +08:00
imported deepseed twice. Removed unnecesary extra import
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user