l the Flax specific dependencies:


	Copied
pip install -U -r requirements_flax.txt
Specify the MODEL_NAME environment variable (either a Hub model repository id or a path to the directory containing the model weights) and pass it to the ~diffusers.DiffusionPipeline.from_pretrained.pretrained_model_name_or_path argument.
Then you can launch the training script:


	Copied
export MODEL_NAME="duongna/stable-diffusion-v1-4-flax"
export DATA_DIR="./