fix: outputs may be tuple

OpenBMB · Dec 15, 2022 · 1074c8a · 1074c8a
1 parent 10d192c
commit 1074c8a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/bmtrain/block_layer.py b/bmtrain/block_layer.py
@@ -49,6 +49,8 @@ def forward(ctx, placeholder, block : 'CheckpointBlock', preserve_rng_state, len
 
         if not isinstance(outputs, list) and not isinstance(outputs, tuple):
             outputs = [outputs]
+        else:
+            outputs = list(outputs)
         return tuple([len(outputs)] + outputs + [hidden_state["tensor"] for hidden_state in inspector.hidden_states])
 
     @staticmethod