From 0afe9072039217e197b8fe527ba0f89c953176b0 Mon Sep 17 00:00:00 2001 From: Eugene-hu <85906264+Eugene-hu@users.noreply.github.com> Date: Wed, 7 Sep 2022 09:18:41 -0700 Subject: [PATCH] local train bug fix (#906) --- bittensor/_neuron/text/core_server/nucleus_impl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bittensor/_neuron/text/core_server/nucleus_impl.py b/bittensor/_neuron/text/core_server/nucleus_impl.py index f2d7d2c19b..1351e98a75 100644 --- 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()