Update alphafold3.py (#211)

Author: amorehead

alphafold3_pytorch/alphafold3.py

@@ -4320,7 +4320,7 @@ def inner(inputs):
         for layer in wrapped_layers:
             inputs = checkpoint(layer, inputs)
 
-        logits, _ = inputs
+        logits, *_ = inputs
         return logits
 
     @typecheck