diffusers
d27e996c - Adding VQGAN Training script (#5483)

Comment changes are shownComment changes are hidden
Commit
1 year ago
Adding VQGAN Training script (#5483) * 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: Sayak 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: Sayak 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: Sayak Paul <spsayakpaul@gmail.com> Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>
Author
Parents
  • .github/workflows
    • File
      pr_tests.yml
    • File
      push_tests.yml
    • File
      push_tests_fast.yml
  • examples/vqgan
    • File
      README.md
    • File
      discriminator.py
    • File
      requirements.txt
    • File
      test_vqgan.py
    • File
      train_vqgan.py
  • src/diffusers
    • models
      • autoencoders
        • File
          vae.py
      • File
        vq_model.py
    • utils
      • File
        __init__.py
      • File
        import_utils.py
      • File
        testing_utils.py
  • tests/models/autoencoders
    • File
      test_models_vq.py