ing script.
xFormers is not available for Flax.

Upload model to Hub
	
Store your model on the Hub by adding the following argument to the training script:


	Copied
  --push_to_hub

Save and load checkpoints
	
It is a good idea to regularly save checkpoints in case anything happens during training. To save a checkpoint, pass the following argument to the training script:


	Copied
  --checkpointing_steps=500
Every 500 steps, the full training 