1. 08 Feb, 2024 1 commit
  2. 31 Jan, 2024 2 commits
  3. 28 Jan, 2024 1 commit
  4. 27 Jan, 2024 1 commit
  5. 26 Jan, 2024 1 commit
    • dg845's avatar
      Add Community Example Consistency Training Script (#6717) · b4220e97
      dg845 authored
      
      
      * initial commit for unconditional/class-conditional consistency training script
      
      * make style
      
      * Add entry for consistency training script in community README.
      
      * Move consistency training script from community to research_projects/consistency_training
      
      * Add requirements.txt and README to research_projects/consistency_training directory.
      
      * Manually revert community README changes for consistency training.
      
      * Fix path to script after moving script to research projects.
      
      * Add option to load U-Net weights from pretrained model.
      
      ---------
      Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
      b4220e97
  6. 24 Jan, 2024 1 commit
  7. 23 Jan, 2024 1 commit
    • Sayak Paul's avatar
      [Big refactor] move unets to `unets` module 🦋 (#6630) · 1f0705ad
      Sayak Paul authored
      * move unets to  module 🦋
      
      * parameterize unet-level import.
      
      * fix flax unet2dcondition model import
      
      * models __init__
      
      * mildly depcrecating models.unet_2d_blocks in favor of models.unets.unet_2d_blocks.
      
      * noqa
      
      * correct depcrecation behaviour
      
      * inherit from the actual classes.
      
      * Empty-Commit
      
      * backwards compatibility for unet_2d.py
      
      * backward compatibility for unet_2d_condition
      
      * bc for unet_1d
      
      * bc for unet_1d_blocks
      1f0705ad
  8. 22 Jan, 2024 1 commit
  9. 15 Jan, 2024 1 commit
  10. 12 Jan, 2024 1 commit
  11. 10 Jan, 2024 1 commit
    • Rahul Raman's avatar
      example: Train Instruct pix2 pix with lora implementation (#6469) · 2d1f2182
      Rahul Raman authored
      
      
      * base template file - train_instruct_pix2pix.py
      
      * additional import and parser argument requried for lora
      
      * finetune only instructpix2pix model -- no need to include these layers
      
      * inject lora layers
      
      * freeze unet model -- only lora layers are trained
      
      * training modifications to train only lora parameters
      
      * store only lora parameters
      
      * move train script to research project
      
      * run quality and style code checks
      
      * move train script to a new folder
      
      * add README
      
      * update README
      
      * update references in README
      
      ---------
      Co-authored-by: default avatarRahul Raman <rahulraman@gmail.com>
      Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
      2d1f2182
  12. 05 Jan, 2024 2 commits
    • Sayak Paul's avatar
      Rename REAMDE.md to README.md · a483a8ed
      Sayak Paul authored
      a483a8ed
    • Sayak Paul's avatar
      [Experimental] Diffusion LoRA DPO training (#6422) · 2a97067b
      Sayak Paul authored
      * add: experimental script for diffusion dpo training.
      
      * random_crop cli.
      
      * fix: caption tokenization.
      
      * fix: pixel_values index.
      
      * fix: grad?
      
      * debug
      
      * fix: reduction.
      
      * fixes in the loss calculation.
      
      * style
      
      * fix: unwrap call.
      
      * fix: validation inference.
      
      * add: initial sdxl script
      
      * debug
      
      * make sure images in the tuple are of same res
      
      * fix model_max_length
      
      * report print
      
      * boom
      
      * fix: numerical issues.
      
      * fix: resolution
      
      * comment about resize.
      
      * change the order of the training transformation.
      
      * save call.
      
      * debug
      
      * remove print
      
      * manually detaching necessary?
      
      * use the same vae for validation.
      
      * add: readme.
      2a97067b
  13. 03 Jan, 2024 1 commit
  14. 29 Dec, 2023 2 commits
  15. 27 Dec, 2023 1 commit
  16. 21 Dec, 2023 1 commit
  17. 11 Dec, 2023 1 commit
  18. 06 Dec, 2023 1 commit
  19. 20 Nov, 2023 1 commit
  20. 17 Nov, 2023 1 commit
  21. 13 Nov, 2023 2 commits
  22. 30 Oct, 2023 1 commit
  23. 16 Oct, 2023 2 commits
  24. 05 Oct, 2023 1 commit
  25. 03 Oct, 2023 1 commit
  26. 02 Oct, 2023 1 commit
  27. 27 Sep, 2023 1 commit
  28. 26 Sep, 2023 1 commit
  29. 20 Sep, 2023 1 commit
  30. 08 Sep, 2023 1 commit
  31. 04 Sep, 2023 1 commit
    • Isamu Isozaki's avatar
      Retrieval Augmented Diffusion Models (#3297) · 3201903d
      Isamu Isozaki authored
      
      
      * Resetting rdm pr
      
      * Fixed styles
      
      * Fixed style
      
      * Moved to rdm folder+fixed slight errors
      
      * Removed config diff
      
      * Started adding tests
      
      * Adding retrieved images
      
      * Fixed faiss import
      
      * Fixed import errors
      
      * Fixing tests
      
      * Added require_faiss
      
      * Updated dependency table
      
      * Attempt solving consistency test
      
      * Fixed truncation and vocab size issue
      
      * Passed common tests
      
      * Finished up cpu testing on pipeline
      
      * Passed all tests locally
      
      * Removed some slow tests
      
      * Removed diffs from test_pipeline_common
      
      * Remove logs
      
      * Removed diffs from test_pipelines_common
      
      * Fixed style
      
      * Fully fixed styles on diffs
      
      * Fixed name
      
      * Proper rename
      
      * Fixed dummies
      
      * Fixed issue with dummyonnx
      
      * Fixed black style
      
      * Fixed dummies
      
      * Changed ordering
      
      * Fixed logging
      
      * Fixing
      
      * Fixing
      
      * quality
      
      * Debugging regex
      
      * Fix dummies with guess
      
      * Fixed typo
      
      * Attempt fix dummies
      
      * black
      
      * ruff
      
      * fixed ordering
      
      * Logging
      
      * Attempt fix
      
      * Attempt fix dummy
      
      * Attempt fixing styles
      
      * Fixed faiss dependency
      
      * Removed unnecessary deprecations
      
      * Finished up main changes
      
      * Added doc
      
      * Passed tests
      
      * Fixed tests
      
      * Remove invisible watermark
      
      * Fixed ruff errors
      
      * Added prompt embed to tests
      
      * Added tests and made retriever an optional component
      
      * Fixed styles
      
      * Made faiss a dependency of pipeline
      
      * Logging
      
      * Fixed dummies
      
      * Make pipeline test work
      
      * Fixed style
      
      * Moved to research projects
      
      * Remove diff
      
      * Fixed style error
      
      ---------
      Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
      3201903d
  32. 16 Aug, 2023 2 commits
  33. 12 Aug, 2023 1 commit
  34. 13 Jul, 2023 1 commit