bug fix in chunking and test

Author: shenoynikhil

src/boltz/model/layers/outer_product_mean.py

@@ -83,7 +83,8 @@ def forward(self, m: Tensor, mask: Tensor, chunk_size: int = None) -> Tensor:
                     z_out = z.to(m) @ sliced_weight_proj_o.T
                 else:
                     z_out = z_out + z.to(m) @ sliced_weight_proj_o.T
-            # z_out = z_out + self.proj_o.bias # add bias
+            
+            z_out = z_out + self.proj_o.bias # add bias
             return z_out
         else:
             mask = mask[:, :, None, :] * mask[:, :, :, None]