2 Commits

Author	SHA1	Message	Date
Yuqian Hong	fc28791fc8	[BUG] Fix Autoencoderkl train script (#11113) ... * add disc_optimizer step (not fix) * support syncbatchnorm in discriminator	2025-03-19 16:49:02 +05:30
Yuqian Hong	4fa24591a3	create a script to train autoencoderkl (#10605) ... * create a script to train vae * update main.py * update train_autoencoderkl.py * update train_autoencoderkl.py * add a check of --pretrained_model_name_or_path and --model_config_name_or_path * remove the comment, remove diffusers in requiremnets.txt, add validation_image ote * update autoencoderkl.py * quality --------- Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>	2025-01-27 16:41:34 +05:30