• Yuqian Hong's avatar
    create a script to train autoencoderkl (#10605) · 4fa24591
    Yuqian Hong authored
    
    
    * 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: default avatarSayak Paul <spsayakpaul@gmail.com>
    4fa24591
README.md 1.67 KB