Commit Graph

10 Commits

Author SHA1 Message Date
Anton Lozhkov eeb9264acd
Support training with a local image folder (#152)
* Support training with an image folder

* style
2022-08-03 15:25:00 +02:00
Anton Lozhkov cde0ed162a
Add a step about accelerate config to the examples (#130) 2022-07-22 13:48:26 +02:00
John Haugeland 85244d4a59
Documentation cross-reference (#127)
In https://github.com/huggingface/diffusers/issues/124 I incorrectly suggested that the image set creation process was undocumented.  In reality, I just hadn't located it.  @patrickvonplaten did so for me.

This PR places a hotlink so that people like me can be shoehorned over where they needed to be.
2022-07-21 21:46:15 +02:00
Anton Lozhkov a487b5095a
Update images 2022-07-21 17:11:36 +02:00
anton-l 302b86bd0b Adapt training to the new UNet API 2022-07-21 11:07:21 +02:00
Anton Lozhkov 76f9b52289
Update the training examples (#102)
* New unet, gradient accumulation

* Save every n epochs

* Remove find_unused_params, hooray!

* Update examples

* Switch to DDPM completely
2022-07-20 19:51:23 +02:00
Anton Lozhkov 33abc79515
Update README.md 2022-06-22 13:52:45 +02:00
anton-l a2117cb797 add push_to_hub 2022-06-21 10:38:34 +02:00
Manuel Romero 57aba1ef50
Fix output path name 2022-06-15 21:45:49 +02:00
Anton Lozhkov 1112699149
add a training examples doc 2022-06-15 16:51:37 +02:00