raulc0399 commited on
Commit
2f5e5d4
·
verified ·
1 Parent(s): c48b54f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
README.md CHANGED
@@ -28,4 +28,8 @@ the weights were created using the controlnet/convert_pixart_alpha_controlnet_to
28
  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
29
  ```
30
 
31
- training script is in the same repo, but is WIP
 
 
 
 
 
28
  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
29
  ```
30
 
31
+ the training script has been implemented in controlnet/train_pixart_controlnet_hf.py
32
+ to start it with the toy-dataset "fusing/fill50k" run the file ./train_controlnet_hf_diffusers.sh
33
+
34
+ more info can be found in this PR
35
+ https://github.com/PixArt-alpha/PixArt-alpha/pull/164