@@ -584,7 +584,6 @@ class BloomForSequenceClassification(BloomPreTrainedModel):
)
pooled_logits = logits[torch.arange(batch_size, device=logits.device), sequence_lengths]
-
loss = None
if labels is not None:
if self.config.problem_type is None: