"vscode:/vscode.git/clone" did not exist on "40e0827e60e2c5b5a99176c2abc1244e77f064b1"
  • 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
train_autoencoderkl.py 42.1 KB