Fix bug

Created using spr 1.3.4
google · Jan 6, 2025 · 4e3ef26 · 4e3ef26
1 parent b6557b6
commit 4e3ef26
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gematria/model/python/model_base.py b/gematria/model/python/model_base.py
@@ -472,7 +472,7 @@ def __call__(self, feed_dict, train=False):
           feed_dict['delta_block_index'],
           name=ModelBase.OUTPUT_TENSOR_NAME,
       )
-      output_dict['output_deltas'] = feed_dict['output_deltas']
+      output_dict['output_deltas'] = output['output_deltas']
 
       return output_dict
     else: