"vscode:/vscode.git/clone" did not exist on "50296739878f3e17b2d25d45ef626318b44440b9"
  • Isamu Isozaki's avatar
    Adding VQGAN Training script (#5483) · d27e996c
    Isamu Isozaki authored
    
    
    * Init commit
    
    * Removed einops
    
    * Added default movq config for training
    
    * Update explanation of prompts
    
    * Fixed inheritance of discriminator and init_tracker
    
    * Fixed incompatible api between muse and here
    
    * Fixed output
    
    * Setup init training
    
    * Basic structure done
    
    * Removed attention for quick tests
    
    * Style fixes
    
    * Fixed vae/vqgan styles
    
    * Removed redefinition of wandb
    
    * Fixed log_validation and tqdm
    
    * Nothing commit
    
    * Added commit loss to lookup_from_codebook
    
    * Update src/diffusers/models/vq_model.py
    Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
    
    * Adding perliminary README
    
    * Fixed one typo
    
    * Local changes
    
    * Fixed main issues
    
    * Merging
    
    * Update src/diffusers/models/vq_model.py
    Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
    
    * Testing+Fixed bugs in training script
    
    * Some style fixes
    
    * Added wandb to docs
    
    * Fixed timm test
    
    * get testing suite ready.
    
    * remove return loss
    
    * remove return_loss
    
    * Remove diffs
    
    * Remove diffs
    
    * fix ruff format
    
    ---------
    Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
    Co-authored-by: default avatarDhruv Nair <dhruv.nair@gmail.com>
    d27e996c
vae.py 35.7 KB