Fix rcps channel splitting bug

Fixes issue identified here: https://github.com/kuleshov-group/caduceus/issues/5

modeling_caduceus.py

@@ -543,8 +543,8 @@ class CaduceusForSequenceClassification(CaduceusPreTrainedModel):
             )
             hidden_states = torch.stack(
                 [
-                    transformer_outputs[0][..., :self.config.d_model // 2],
-                    torch.flip(transformer_outputs[0][..., self.config.d_model // 2:], dims=[1, 2])
+                    transformer_outputs[0][..., :self.config.d_model],
+                    torch.flip(transformer_outputs[0][..., self.config.d_model:], dims=[1, 2])
                  ],
                 dim=-1
             )