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 |
|