remove invalid line in hf-models (deepchem#3866)

GreatRSingh · Mar 5, 2024 · c6a33c2 · c6a33c2
1 parent d84bc6e
commit c6a33c2
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/deepchem/models/torch_models/hf_models.py b/deepchem/models/torch_models/hf_models.py
@@ -330,8 +330,6 @@ def fit_generator(self,
             optimizer.zero_grad()
             outputs = self.model(**inputs)
 
-            if self._loss_outputs is not None:
-                outputs = [outputs[i] for i in self._loss_outputs]
             batch_loss = outputs.get("loss")
             batch_loss.backward()
             optimizer.step()