Skip to content

v2.6

Latest

Choose a tag to compare

@IrisRainbowNeko IrisRainbowNeko released this 04 Jan 12:34
  • WebDSText2ImageSource: images in webdataset and text captions in label_file (.txt, .json, ...)
  • Add drop probability option to TagDropoutHandler
  • Fix GWLoss
  • Remake sampler
  • Add solver to sampler (Eular ODE solver, DPM dolver, ...)
  • Fix bugs of attention_mask in TokenizeHandler
  • Preview images with logger (preview to wandb/tensorboard, supported by rainbowneko>=1.10)
  • Cache trainable_parameters in wrapper to reduce overhead. (Iteration over trainable_parameters is time-consuming)
  • Support VAEs with latents_mean and latents_std