Update README.md

README.md

@@ -15,6 +15,7 @@ To use:
 
 ```bash
 git clone https://github.com/raulc0399/PixArt-alpha.git .
+git checkout master_train_controlnet_diffusers
 pip install -r requirements.txt
 python controlnet/pipeline/run_controlnet_pipeline.py
 ```
@@ -25,4 +26,6 @@ the weights were created using the controlnet/convert_pixart_alpha_controlnet_to
 
 ```bash
 python controlnet/convert_pixart_alpha_controlnet_to_diffusers.py --orig_ckpt_path=<full path to>/PixArt-XL-2-1024-ControlNet.pth --dump_path=./<output folder>/hed-controlnet
-```
+```
+
+training script is in the same repo, but is WIP