 it doesn’t support gradient checkpointing or xFormers. You should have a GPU with >30GB of memory if you want to train faster with Flax. This guide will explore the train_controlnet.py training script to help you become familiar with it, and how you can adapt it for your own use-case. Before running the script, make sure you install the library from source:   Copied git clone https://github.com/huggingface/diffusers
cd diffusers
pip install . 