Fix ValueError when save model using `.save_pretrained` due to the non-contiguous tensor in Midm model

modeling_midm.py

@@ -460,6 +460,18 @@ class MidmPreTrainedModel(PreTrainedModel):
         if isinstance(module, MidmModel):
             module.gradient_checkpointing = value
 
+    def make_tensors_contiguous(self):
+        for name, param in self.named_parameters():
+            if not param.is_contiguous():
+                param.data = param.data.contiguous()
+
+    def save_pretrained(self, save_directory, **kwargs):
+        # Make tensors contiguous
+        self.make_tensors_contiguous()
+
+        # Call the original save_pretrained method
+        super().save_pretrained(save_directory, **kwargs)
+
 
 @dataclass
 class MidmDoubleHeadsModelOutput(ModelOutput):