Răsfoiți Sursa

revert changes

dbaranchuk 3 ani în urmă
părinte
comite
0e8b7490f7
1 a modificat fișierele cu 0 adăugiri și 1 ștergeri
  1. 0 1
      src/bloom/model.py

+ 0 - 1
src/bloom/model.py

@@ -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: