1. 04 Mar, 2024 1 commit
    • Linoy Tsaban's avatar
      [training scripts] add tags of diffusers-training (#7206) · 8da360aa
      Linoy Tsaban authored
      * add tags for diffusers training
      
      * add tags for diffusers training
      
      * add tags for diffusers training
      
      * add tags for diffusers training
      
      * add tags for diffusers training
      
      * add tags for diffusers training
      
      * add dora tags for drambooth lora scripts
      
      * style
      8da360aa
  2. 28 Feb, 2024 1 commit
  3. 16 Feb, 2024 1 commit
  4. 09 Feb, 2024 1 commit
  5. 08 Feb, 2024 1 commit
  6. 31 Jan, 2024 2 commits
  7. 28 Jan, 2024 1 commit
  8. 27 Jan, 2024 1 commit
  9. 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
  10. 24 Jan, 2024 1 commit
  11. 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
  12. 22 Jan, 2024 1 commit
  13. 15 Jan, 2024 1 commit
  14. 12 Jan, 2024 1 commit
  15. 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
  16. 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
  17. 03 Jan, 2024 1 commit
  18. 29 Dec, 2023 2 commits
  19. 27 Dec, 2023 1 commit
  20. 21 Dec, 2023 1 commit
  21. 11 Dec, 2023 1 commit
  22. 06 Dec, 2023 1 commit
  23. 20 Nov, 2023 1 commit
  24. 17 Nov, 2023 1 commit
  25. 13 Nov, 2023 2 commits
  26. 30 Oct, 2023 1 commit
  27. 16 Oct, 2023 2 commits
  28. 05 Oct, 2023 1 commit
  29. 03 Oct, 2023 1 commit
  30. 02 Oct, 2023 1 commit
  31. 27 Sep, 2023 1 commit
  32. 26 Sep, 2023 1 commit
  33. 20 Sep, 2023 1 commit
  34. 08 Sep, 2023 1 commit
  35. 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