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
a73ae3e5b0
Better default for AdamW
2022-07-21 13:36:16 +02:00
anton-l
06505ba4b4
Less eval steps during training
2022-07-21 11:47:40 +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
d9316bf8bc
Fix mutable proj_out weight in the Attention layer ( #73 )
...
* Catch unused params in DDP
* Fix proj_out, add test
2022-07-04 12:36:37 +02:00
Tanishq Abraham
3abf4bc439
EMA model stepping updated to keep track of current step ( #64 )
...
ema model stepping done automatically now
2022-07-04 11:53:15 +02:00
Anton Lozhkov
8cba133f36
Add the model card template ( #43 )
...
* add a metrics logger
* fix LatentDiffusionUncondPipeline
* add VQModel in init
* add image logging to tensorboard
* switch manual templates to the modelcards package
* hide ldm example
Co-authored-by: patil-suraj <surajp815@gmail.com>
2022-06-29 15:37:23 +02:00
Patrick von Platen
932ce05d97
cancel einops
2022-06-27 15:39:41 +00:00
anton-l
07ff0abff4
Glide and LDM training experiments
2022-06-27 17:25:59 +02:00
anton-l
1cf7933ea2
Framework-agnostic timestep broadcasting
2022-06-27 17:11:01 +02:00
anton-l
3f9e3d8ad6
add EMA during training
2022-06-27 15:23:01 +02:00
anton-l
c31736a4a4
Merge remote-tracking branch 'origin/main'
...
# Conflicts:
# src/diffusers/pipelines/pipeline_glide.py
2022-06-22 15:17:10 +02:00
anton-l
7b43035bcb
init text2im script
2022-06-22 15:15:54 +02:00
Anton Lozhkov
33abc79515
Update README.md
2022-06-22 13:52:45 +02:00
anton-l
848c86ca0a
batched forward diffusion step
2022-06-22 13:38:14 +02:00
anton-l
9e31c6a749
refactor GLIDE text2im pipeline, remove classifier_free_guidance
2022-06-21 14:07:58 +02:00
anton-l
71289ba06e
add lr schedule utils
2022-06-21 11:35:56 +02:00
anton-l
0417baf23d
additional hub arguments
2022-06-21 11:21:10 +02:00
anton-l
9c82c32ba7
make style
2022-06-21 10:43:40 +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
Patrick von Platen
fb9e37adf6
correct logging
2022-06-15 15:52:23 +02:00
anton-l
84bd65bced
Merge remote-tracking branch 'origin/main'
2022-06-15 14:37:04 +02:00
anton-l
0deeb06aac
better defaults
2022-06-15 14:36:43 +02:00
Patrick von Platen
17c574a16d
remove torchvision dependency
2022-06-15 12:35:47 +02:00
anton-l
cfe6eb1611
Training example parameterization
2022-06-15 11:21:02 +02:00
anton-l
7fe05bb311
Bugfixes for the training example
2022-06-14 18:25:22 +02:00
anton-l
bb30664285
Move the training example
2022-06-14 11:33:24 +02:00
Patrick von Platen
3b8f24525f
upload
2022-06-07 13:24:36 +00:00
Patrick von Platen
fe3137304b
improve
2022-06-06 17:03:41 +02:00
Patrick von Platen
3a5c65d568
finish
2022-06-03 19:11:58 +02:00
Patrick von Platen
417927f554
add some examples to seperate sampler and schedules
2022-06-03 19:02:36 +02:00