• Andrés Mauricio Repetto Ferrero's avatar
    Adding better way to define multiple concepts and also validation capabilities. (#3807) · 572d8e20
    Andrés Mauricio Repetto Ferrero authored
    
    
    * - Added validation parameters
    - Changed some parameter descriptions to better explain their use.
    - Fixed a few typos.
    - Added concept_list parameter for better management of multiple subjects
    - changed logic for image validation
    
    * - Fixed bad logic for class data root directories
    
    * Defaulting validation_steps to None for an easier logic
    
    * Fixed multiple validation prompts
    
    * Fixed bug on validation negative prompt
    
    * Changed validation logic for tracker.
    
    * Added uuid for validation image labeling
    
    * Fix error when comparing validation prompts and validation negative prompts
    
    * Improved error message when negative prompts for validation are more than the number of prompts
    
    * - Changed image tracking number from epoch to global_step
    - Added Typing for functions
    
    * Added some validations more when using concept_list parameter and the regular ones.
    
    * Fixed error message
    
    * Added more validations for validation parameters
    
    * Improved messaging for errors
    
    * Fixed validation error for parameters with default values
    
    * - Added train step to image name for validation
    - reformatted code
    
    * - Added train step to image's name for validation
    - reformatted code
    
    * Updated README.md file.
    
    * reverted back original script of train_dreambooth.py
    
    * reverted back original script of train_dreambooth.py
    
    * left one blank line at the eof
    
    * reverted back setup.py
    
    * reverted back setup.py
    
    * added same logic for when parameters for prior preservation are used without enabling the flag while using concept_list parameter.
    
    * Ran black formatter.
    
    * fixed a few strings
    
    * fixed import sort with isort and removed fstrings without placeholder
    
    * fixed import order with ruff (since with isort wasn't ok)
    
    ---------
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    572d8e20
README.md 14.4 KB