bugfix: tuple -> list

SeanLee97 · Jan 11, 2025 · 18af7f3 · 18af7f3
1 parent 07f236b
commit 18af7f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/angle_emb/angle.py b/angle_emb/angle.py
@@ -716,7 +716,7 @@ def __call__(self,
             outputs = ret.last_hidden_state
         else:
             ret = self.model(output_hidden_states=True, return_dict=True, **inputs)
-            all_layer_outputs = ret.hidden_states
+            all_layer_outputs = list(ret.hidden_states)
             all_layer_outputs[-1] = ret.last_hidden_state
             if return_all_layer_outputs:
                 return (all_layer_outputs, ret.logits) if return_mlm_logits else all_layer_outputs