local train bug fix (#906)

opentensor · Sep 7, 2022 · 0afe907 · 0afe907
1 parent a778fb8
commit 0afe907
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bittensor/_neuron/text/core_server/nucleus_impl.py b/bittensor/_neuron/text/core_server/nucleus_impl.py
@@ -234,7 +234,7 @@ def forward(self, inputs, tokenizer=None):
                     Decoded predictions of the next token in the sentence.
 
         """
-        message, model_output, decoded_targets = self.local_forward(inputs, tokenizer)[1]
+        message, model_output, decoded_targets = self.local_forward(inputs, tokenizer)
 
         shift_logits = decoded_targets[..., :-1, :].contiguous()
         shift_labels = inputs[..., 1:].contiguous()