From 1074c8a2b5e902245e9e0b475b18f855f744a90c Mon Sep 17 00:00:00 2001
From: William <acha131441373@gmail.com>
Date: Thu, 15 Dec 2022 16:35:53 +0800
Subject: [PATCH] fix: outputs may be tuple

---
 bmtrain/block_layer.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/bmtrain/block_layer.py b/bmtrain/block_layer.py
index a823c877..8bde0ab2 100644
--- 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