Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Version 0.8.6 #1212

Merged
merged 98 commits into from
Apr 7, 2024
Merged

Version 0.8.6 #1212

merged 98 commits into from
Apr 7, 2024

Commits on Feb 3, 2024

  1. support deepspeed

    BootsofLagrangian committed Feb 3, 2024
    Configuration menu
    Copy the full SHA
    dfe08f3 View commit details
    Browse the repository at this point in the history

Commits on Feb 5, 2024

  1. Configuration menu
    Copy the full SHA
    64873c1 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    2824312 View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    4295f91 View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    3970bf4 View commit details
    Browse the repository at this point in the history
  5. Configuration menu
    Copy the full SHA
    7d2a926 View commit details
    Browse the repository at this point in the history

Commits on Feb 7, 2024

  1. Configuration menu
    Copy the full SHA
    6255661 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    2445a5b View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    a98feca View commit details
    Browse the repository at this point in the history

Commits on Feb 9, 2024

  1. Configuration menu
    Copy the full SHA
    03f0816 View commit details
    Browse the repository at this point in the history

Commits on Feb 22, 2024

  1. Configuration menu
    Copy the full SHA
    4d5186d View commit details
    Browse the repository at this point in the history

Commits on Feb 26, 2024

  1. 2 Configuration menu
    Copy the full SHA
    577e991 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    f2c727f View commit details
    Browse the repository at this point in the history
  3. update readme

    kohya-ss committed Feb 26, 2024
    Configuration menu
    Copy the full SHA
    1751936 View commit details
    Browse the repository at this point in the history
  4. fix typo

    kohya-ss committed Feb 26, 2024
    Configuration menu
    Copy the full SHA
    4a5546d View commit details
    Browse the repository at this point in the history

Commits on Feb 27, 2024

  1. Merge branch 'main' into dev

    kohya-ss committed Feb 27, 2024
    Configuration menu
    Copy the full SHA
    074d32a View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    eefb3cc View commit details
    Browse the repository at this point in the history
  3. Merge pull request #1101 from BootsofLagrangian/deepspeed

    support deepspeed
    kohya-ss authored Feb 27, 2024
    Configuration menu
    Copy the full SHA
    0e4a573 View commit details
    Browse the repository at this point in the history
  4. make deepspeed_utils

    kohya-ss committed Feb 27, 2024
    Configuration menu
    Copy the full SHA
    e3ccf8f View commit details
    Browse the repository at this point in the history
  5. Configuration menu
    Copy the full SHA
    a9b64ff View commit details
    Browse the repository at this point in the history

Commits on Mar 3, 2024

  1. Configuration menu
    Copy the full SHA
    14c9372 View commit details
    Browse the repository at this point in the history

Commits on Mar 10, 2024

  1. Add "encoding='utf-8'"

    Horizon1704 authored Mar 10, 2024
    Configuration menu
    Copy the full SHA
    124ec45 View commit details
    Browse the repository at this point in the history
  2. save state on train end

    gesen2egee committed Mar 10, 2024
    Configuration menu
    Copy the full SHA
    095b803 View commit details
    Browse the repository at this point in the history

Commits on Mar 11, 2024

  1. Update train_network.py

    gesen2egee committed Mar 11, 2024
    Configuration menu
    Copy the full SHA
    d282c45 View commit details
    Browse the repository at this point in the history

Commits on Mar 12, 2024

  1. Configuration menu
    Copy the full SHA
    74c266a View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    97524f1 View commit details
    Browse the repository at this point in the history

Commits on Mar 13, 2024

  1. Configuration menu
    Copy the full SHA
    948029f View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    8639940 View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    53954a1 View commit details
    Browse the repository at this point in the history

Commits on Mar 15, 2024

  1. Merge branch 'main' into dev

    kohya-ss committed Mar 15, 2024
    Configuration menu
    Copy the full SHA
    0a8ec52 View commit details
    Browse the repository at this point in the history
  2. fix doc

    kohya-ss committed Mar 15, 2024
    Configuration menu
    Copy the full SHA
    443f029 View commit details
    Browse the repository at this point in the history

Commits on Mar 17, 2024

  1. Configuration menu
    Copy the full SHA
    0ef4fe7 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    7081a0c View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    3419c3d View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    86e40fa View commit details
    Browse the repository at this point in the history

Commits on Mar 18, 2024

  1. Update tag_images_by_wd14_tagger.py

    add WDV3
    sdbds committed Mar 18, 2024
    Configuration menu
    Copy the full SHA
    a7dff59 View commit details
    Browse the repository at this point in the history
  2. Update requirements.txt

    sdbds committed Mar 18, 2024
    Configuration menu
    Copy the full SHA
    5410a8c View commit details
    Browse the repository at this point in the history
  3. Update requirements.txt

    sdbds committed Mar 18, 2024
    Configuration menu
    Copy the full SHA
    a71c35c View commit details
    Browse the repository at this point in the history

Commits on Mar 20, 2024

  1. fix typo

    sdbds committed Mar 20, 2024
    Configuration menu
    Copy the full SHA
    6c51c97 View commit details
    Browse the repository at this point in the history
  2. Merge branch 'main' into dev

    kohya-ss committed Mar 20, 2024
    Configuration menu
    Copy the full SHA
    e281e86 View commit details
    Browse the repository at this point in the history
  3. Merge pull request #1192 from sdbds/main

    Add WDV3 support
    kohya-ss authored Mar 20, 2024
    Configuration menu
    Copy the full SHA
    7da41be View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    80dbbf5 View commit details
    Browse the repository at this point in the history
  5. Merge pull request #1177 from KohakuBlueleaf/random-strength-noise

    Random strength for Noise Offset and input perturbation noise
    kohya-ss authored Mar 20, 2024
    Configuration menu
    Copy the full SHA
    cf09c6a View commit details
    Browse the repository at this point in the history
  6. English Translation of config_README-ja.md (#1175)

    * Add files via upload
    
    Creating template to work on.
    
    * Update config_README-en.md
    
    Total Conversion from Japanese to English.
    
    * Update config_README-en.md
    
    * Update config_README-en.md
    
    * Update config_README-en.md
    darkstorm2150 authored Mar 20, 2024
    Configuration menu
    Copy the full SHA
    46331a9 View commit details
    Browse the repository at this point in the history
  7. update readme

    kohya-ss committed Mar 20, 2024
    Configuration menu
    Copy the full SHA
    5f6196e View commit details
    Browse the repository at this point in the history
  8. Merge pull request #1167 from Horizon1704/patch-1

    Add "encoding='utf-8'" for --config_file
    kohya-ss authored Mar 20, 2024
    Configuration menu
    Copy the full SHA
    119cc99 View commit details
    Browse the repository at this point in the history
  9. update readme

    kohya-ss committed Mar 20, 2024
    Configuration menu
    Copy the full SHA
    3b0db0f View commit details
    Browse the repository at this point in the history
  10. Merge pull request #1168 from gesen2egee/save_state_on_train_end

    Save state on train end
    kohya-ss authored Mar 20, 2024
    Configuration menu
    Copy the full SHA
    bf6cd4b View commit details
    Browse the repository at this point in the history
  11. Configuration menu
    Copy the full SHA
    855add0 View commit details
    Browse the repository at this point in the history
  12. Configuration menu
    Copy the full SHA
    9b6b39f View commit details
    Browse the repository at this point in the history
  13. Configuration menu
    Copy the full SHA
    fbb98f1 View commit details
    Browse the repository at this point in the history
  14. Fix most of ZeRO stage uses optimizer partitioning

     - we have to prepare optimizer and ds_model at the same time.
     - pull/1139#issuecomment-1986790007
    
    Signed-off-by: BootsofLagrangian <[email protected]>
    BootsofLagrangian committed Mar 20, 2024
    Configuration menu
    Copy the full SHA
    d945602 View commit details
    Browse the repository at this point in the history
  15. Merge pull request #1200 from BootsofLagrangian/deep-speed

    Fix sdxl_train.py in deepspeed branch
    kohya-ss authored Mar 20, 2024
    Configuration menu
    Copy the full SHA
    a35e7bd View commit details
    Browse the repository at this point in the history
  16. update dataset config doc

    kohya-ss committed Mar 20, 2024
    Configuration menu
    Copy the full SHA
    d17c0f5 View commit details
    Browse the repository at this point in the history

Commits on Mar 23, 2024

  1. format by black

    kohya-ss committed Mar 23, 2024
    Configuration menu
    Copy the full SHA
    594c7f7 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    f4a4c11 View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    0c7baea View commit details
    Browse the repository at this point in the history

Commits on Mar 24, 2024

  1. Configuration menu
    Copy the full SHA
    79d1c12 View commit details
    Browse the repository at this point in the history
  2. update readme

    kohya-ss committed Mar 24, 2024
    Configuration menu
    Copy the full SHA
    691f043 View commit details
    Browse the repository at this point in the history
  3. Merge pull request #1205 from feffy380/patch-1

    register reg images with correct subset
    kohya-ss authored Mar 24, 2024
    Configuration menu
    Copy the full SHA
    ad97410 View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    381c449 View commit details
    Browse the repository at this point in the history
  5. [Experimental] Add cache mechanism for dataset groups to avoid long w…

    …aiting time for initilization (#1178)
    
    * support meta cached dataset
    
    * add cache meta scripts
    
    * random ip_noise_gamma strength
    
    * random noise_offset strength
    
    * use correct settings for parser
    
    * cache path/caption/size only
    
    * revert mess up commit
    
    * revert mess up commit
    
    * Update requirements.txt
    
    * Add arguments for meta cache.
    
    * remove pickle implementation
    
    * Return sizes when enable cache
    
    ---------
    
    Co-authored-by: Kohya S <[email protected]>
    KohakuBlueleaf and kohya-ss authored Mar 24, 2024
    Configuration menu
    Copy the full SHA
    ae97c8b View commit details
    Browse the repository at this point in the history
  6. Configuration menu
    Copy the full SHA
    0253472 View commit details
    Browse the repository at this point in the history
  7. Configuration menu
    Copy the full SHA
    8d58588 View commit details
    Browse the repository at this point in the history
  8. Configuration menu
    Copy the full SHA
    993b2ab View commit details
    Browse the repository at this point in the history
  9. format by black

    kohya-ss committed Mar 24, 2024
    Configuration menu
    Copy the full SHA
    1648ade View commit details
    Browse the repository at this point in the history
  10. Configuration menu
    Copy the full SHA
    9bbb28c View commit details
    Browse the repository at this point in the history
  11. Configuration menu
    Copy the full SHA
    9c4492b View commit details
    Browse the repository at this point in the history

Commits on Mar 25, 2024

  1. Configuration menu
    Copy the full SHA
    c24422f View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    a2b8531 View commit details
    Browse the repository at this point in the history

Commits on Mar 26, 2024

  1. Merge pull request #1139 from kohya-ss/deep-speed

    Deep speed
    kohya-ss authored Mar 26, 2024
    Configuration menu
    Copy the full SHA
    ea05e3f View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    ab1e389 View commit details
    Browse the repository at this point in the history
  3. Merge pull request #1207 from kohya-ss/masked-loss

    Add masked loss
    kohya-ss authored Mar 26, 2024
    Configuration menu
    Copy the full SHA
    5a2afb3 View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    c86e356 View commit details
    Browse the repository at this point in the history
  5. Merge pull request #1206 from kohya-ss/dataset-cache

    Add metadata caching for DreamBooth dataset
    kohya-ss authored Mar 26, 2024
    Configuration menu
    Copy the full SHA
    78e0a76 View commit details
    Browse the repository at this point in the history
  6. update readme

    kohya-ss committed Mar 26, 2024
    Configuration menu
    Copy the full SHA
    6c08e97 View commit details
    Browse the repository at this point in the history

Commits on Mar 27, 2024

  1. Configuration menu
    Copy the full SHA
    6f7e93d View commit details
    Browse the repository at this point in the history
  2. Merge pull request #1213 from Disty0/dev

    Add OpenVINO and ROCm ONNX Runtime for WD14
    kohya-ss authored Mar 27, 2024
    Configuration menu
    Copy the full SHA
    b86af67 View commit details
    Browse the repository at this point in the history
  3. Rating support for WD Tagger

    Disty0 committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    dd9763b View commit details
    Browse the repository at this point in the history
  4. fix typo

    Disty0 committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    954731d View commit details
    Browse the repository at this point in the history

Commits on Mar 28, 2024

  1. IPEX fix pin_memory

    Disty0 committed Mar 28, 2024
    Configuration menu
    Copy the full SHA
    4012fd2 View commit details
    Browse the repository at this point in the history

Commits on Mar 29, 2024

  1. Configuration menu
    Copy the full SHA
    bc586ce View commit details
    Browse the repository at this point in the history

Commits on Mar 30, 2024

  1. Configuration menu
    Copy the full SHA
    f1f30ab View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    ae3f625 View commit details
    Browse the repository at this point in the history
  3. update readme

    kohya-ss committed Mar 30, 2024
    Configuration menu
    Copy the full SHA
    434dc40 View commit details
    Browse the repository at this point in the history
  4. Merge pull request #1216 from Disty0/dev

    Rating support for WD Tagger
    kohya-ss authored Mar 30, 2024
    Configuration menu
    Copy the full SHA
    6ba8428 View commit details
    Browse the repository at this point in the history
  5. update wd14 tagger and doc

    kohya-ss committed Mar 30, 2024
    Configuration menu
    Copy the full SHA
    cae5aa0 View commit details
    Browse the repository at this point in the history
  6. update tagger doc

    kohya-ss committed Mar 30, 2024
    Configuration menu
    Copy the full SHA
    f5323e3 View commit details
    Browse the repository at this point in the history
  7. update tagger doc

    kohya-ss committed Mar 30, 2024
    Configuration menu
    Copy the full SHA
    2c2ca9d View commit details
    Browse the repository at this point in the history
  8. fix typo

    kohya-ss committed Mar 30, 2024
    Configuration menu
    Copy the full SHA
    059ee04 View commit details
    Browse the repository at this point in the history

Commits on Mar 31, 2024

  1. Configuration menu
    Copy the full SHA
    2258a1b View commit details
    Browse the repository at this point in the history

Commits on Apr 3, 2024

  1. Configuration menu
    Copy the full SHA
    b748b48 View commit details
    Browse the repository at this point in the history

Commits on Apr 4, 2024

  1. Configuration menu
    Copy the full SHA
    cd587ce View commit details
    Browse the repository at this point in the history

Commits on Apr 7, 2024

  1. Merge pull request #1240 from kohya-ss/verify-command-line-args

    verify command line args if wandb is enabled
    kohya-ss authored Apr 7, 2024
    Configuration menu
    Copy the full SHA
    921036d View commit details
    Browse the repository at this point in the history
  2. update readme

    kohya-ss committed Apr 7, 2024
    Configuration menu
    Copy the full SHA
    089727b View commit details
    Browse the repository at this point in the history
  3. Add option to use Scheduled Huber Loss in all training pipelines to i…

    …mprove resilience to data corruption (#1228)
    
    * add huber loss and huber_c compute to train_util
    
    * add reduction modes
    
    * add huber_c retrieval from timestep getter
    
    * move get timesteps and huber to own function
    
    * add conditional loss to all training scripts
    
    * add cond loss to train network
    
    * add (scheduled) huber_loss to args
    
    * fixup twice timesteps getting
    
    * PHL-schedule should depend on noise scheduler's num timesteps
    
    * *2 multiplier to huber loss cause of 1/2 a^2 conv.
    
    The Taylor expansion of sqrt near zero gives 1/2 a^2, which differs from a^2 of the standard MSE loss. This change scales them better against one another
    
    * add option for smooth l1 (huber / delta)
    
    * unify huber scheduling
    
    * add snr huber scheduler
    
    ---------
    
    Co-authored-by: Kohya S <[email protected]>
    kabachuha and kohya-ss authored Apr 7, 2024
    Configuration menu
    Copy the full SHA
    90b1879 View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    d30ebb2 View commit details
    Browse the repository at this point in the history
  5. update readme

    kohya-ss committed Apr 7, 2024
    Configuration menu
    Copy the full SHA
    dfa3079 View commit details
    Browse the repository at this point in the history