diff --git a/dan/manager/training.py b/dan/manager/training.py
index 7b1423e0c1198b9beccdee2faa0e3d57b15fecb7..72acce307e2ca0f112584753269c1b41847ed74c 100644
--- a/dan/manager/training.py
+++ b/dan/manager/training.py
@@ -1133,9 +1133,7 @@ class Manager(OCRManager):
             ).permute(2, 0, 1)
 
             for i in range(0, max_chars):
-                output, pred, hidden_predict, cache, weights, temperature = self.models[
-                    "decoder"
-                ](
+                output, pred, hidden_predict, cache, weights = self.models["decoder"](
                     features,
                     enhanced_features,
                     predicted_tokens,