src/transformers/integrations/integration_utils.py/DVCLiveCallback/on_save
class DVCLiveCallback:    def on_save(self, args, state, control, **kwargs):
        if self._log_model == "all" and self._initialized and state.is_world_process_zero:
            self.live.log_artifact(args.output_dir)