Code

support Sana training#1187

Code

Merged

bghira merged 18 commits intomainbghira/SimpleTuner:mainfrom feature/sanabghira/SimpleTuner:feature/sanaCopy head branch name to clipboardDec 4, 2024

+820-102

Commits

Commits on Dec 4, 2024

• sana: add to configurator

  

  bghira

  committed
  4106533

  Copy full SHA for 4106533

• sana: text embed caching

  

  bghira

  committed
  3c3b51f

  Copy full SHA for 3c3b51f

• modify vaecache to support Sana return values

  

  bghira

  committed
  e671730

  Copy full SHA for e671730

• add sana to --model_family and enforce "no" mixed_precision

  

  bghira

  committed
  f605790

  Copy full SHA for f605790

• allow exiting on configuration load error

  

  bghira

  committed
  a711d73

  Copy full SHA for a711d73

• add sana configuration safety checks

  

  bghira

  committed
  d01dd32

  Copy full SHA for d01dd32

• sana: model loading

  

  bghira

  committed
  a642d9a

  Copy full SHA for a642d9a

• bugfix for fp16 training and bf16 optim being disabled

  

  bghira

  committed
  12220e4

  Copy full SHA for 12220e4

• savehooks: enable saving and resuming sana models

  

  bghira

  committed
  442d0ea

  Copy full SHA for 442d0ea

• add sana model to list

  

  bghira

  committed
  f513d41

  Copy full SHA for f513d41

• sana: load gemma2 text encoder

  

  bghira

  committed
  0e12e11

  Copy full SHA for 0e12e11

• sana: support for model validations, noise scheduler

  

  bghira

  committed
  f840fe2

  Copy full SHA for f840fe2

• trainer: exit on error when loading config

  

  bghira

  committed
  2fb5764

  Copy full SHA for 2fb5764

• trainer: add sana support, remove --validation_on_startup logic since we have benchmarking now

  

  bghira

  committed
  61e58eb

  Copy full SHA for 61e58eb

• sana: quickstart

  

  bghira

  committed
  baa68db

  Copy full SHA for baa68db

• sana: add to readme

  

  bghira

  committed
  99a9f27

  Copy full SHA for 99a9f27

• add sana to quickstart list

  

  bghira

  committed
  f54e40d

  Copy full SHA for f54e40d

• update diffusers for sana dependencies

  

  bghira

  committed
  bc0b81e

  Copy full SHA for bc0b81e