Code

Better train#6301

Code

Closed

kopyl wants to merge 16 commits intohuggingface:mainhuggingface/diffusers:mainfrom kopyl:better-trainkopyl/diffusers:better-trainCopy head branch name to clipboard

+129-38

Commits

Commits on Dec 12, 2023

• Put deepspeed plugin to a separate function

  kopyl

  committed
  c541989

  Copy full SHA for c541989

• Make saving unet for deepspeed

  kopyl

  committed
  971a4ae

  Copy full SHA for 971a4ae

• Allow specifying saving unet only

  kopyl

  committed
  4fe1d82

  Copy full SHA for 4fe1d82

• Adjust saving unet for multi-GPU

  kopyl

  committed
  9947bd4

  Copy full SHA for 9947bd4

• Make better logging

  kopyl

  committed
  ef2588e

  Copy full SHA for ef2588e

• Make it possible to log images on checkpointing

  kopyl

  committed
  165cdb2

  Copy full SHA for 165cdb2

• Merge pull request #1 from kopyl/better-args

  Show description for f1fde45
  kopyl

  authored
  f1fde45

  Copy full SHA for f1fde45

Commits on Dec 16, 2023

• Utilize my scheduler for validation

  kopyl

  committed
  71e2ed0

  Copy full SHA for 71e2ed0

• Make it possible to control image gen with args

  kopyl

  committed
  9d69afa

  Copy full SHA for 9d69afa

• Merge pull request #2 from kopyl/better-validation

  Show description for 25804de
  kopyl

  authored
  25804de

  Copy full SHA for 25804de

• Make better paths for saving unets and checkpoints

  kopyl

  committed
  753ac87

  Copy full SHA for 753ac87

• Merge pull request #3 from kopyl/better-paths

  Show description for 43b067f
  kopyl

  authored
  43b067f

  Copy full SHA for 43b067f

Commits on Dec 17, 2023

• Switch logging for Docker to log every step

  Show description for 7990bd6
  kopyl

  committed
  7990bd6

  Copy full SHA for 7990bd6

• Merge pull request #4 from kopyl/logging-for-docker

  Show description for fd42e93
  kopyl

  authored
  fd42e93

  Copy full SHA for fd42e93

Commits on Dec 23, 2023

• Make it possible to map data for over 30 minutes

  kopyl

  committed
  29aee80

  Copy full SHA for 29aee80

• Make it possible to use a pre-computed dataset

  kopyl

  committed
  d205a57

  Copy full SHA for d205a57