diffusers
Add extra performance features for EMAModel, torch._foreach operations and better support for non-blocking CPU offloading
#7685
Merged

Commits
  • Add support for _foreach operations and non-blocking to EMAModel
    drhead committed 1 year ago
  • default foreach to false
    drhead committed 1 year ago
  • add non-blocking EMA offloading to SD1.5 T2I example script
    drhead committed 1 year ago
  • fix whitespace
    drhead committed 1 year ago
  • move foreach to cli argument
    drhead committed 1 year ago
  • Merge branch 'main' into patch-1
    sayakpaul committed 1 year ago
  • linting
    drhead committed 1 year ago
  • Update README.md re: EMA weight training
    drhead committed 1 year ago
  • Merge branch 'main' into patch-1
    sayakpaul committed 1 year ago
  • Merge branch 'main' into patch-1
    sayakpaul committed 1 year ago
  • correct args.foreach_ema
    drhead committed 1 year ago
  • Merge branch 'main' into patch-1
    drhead committed 1 year ago
  • add tests for foreach ema
    drhead committed 1 year ago
  • Merge branch 'main' into patch-1
    sayakpaul committed 1 year ago
  • code quality
    sayakpaul committed 1 year ago
  • add foreach to from_pretrained
    drhead committed 1 year ago
  • default foreach false
    drhead committed 1 year ago
  • fix linting
    drhead committed 1 year ago
  • Merge branch 'main' into patch-1
    sayakpaul committed 1 year ago
Loading