*** empty log message ***

modeling_mlp.py

@@ -57,7 +57,7 @@ class MLP(PreTrainedModel):
         # Flatten over seq_length if desired, or do a pooling:
         # Option A: Flatten everything: B x (L*E)
         B, L, E = inputs_embeds.shape
-        assert L == self.config.sequence_length and E == self.config.embedding_size
+        # assert L == self.config.sequence_length and E == self.config.embedding_size
         x = inputs_embeds.reshape(B, L * E)
 
         # Option B: Mean-pool across tokens (comment out if you prefer flattening)