def METHOD_NAME(self, model: nn.Module, checkpoint: str, gather_dtensor: bool, use_safetensors: bool):
    """
    Save model to checkpoint but only on master process.
    """
    if self.coordinator.is_master():
        super().METHOD_NAME(model, checkpoint, gather_dtensor, use_safetensors)