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 |