Commit Graph

702 Commits

Author SHA1 Message Date
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 13457002c0 Merge branch 'main' of github.com:huggingface/diffusers 2022-07-21 11:07:41 +02:00
anton-l 302b86bd0b Adapt training to the new UNet API 2022-07-21 11:07:21 +02:00
Lysandre Debut d87d5edf66
README improvements: credits and roadmap (#116)
* Typos

* Credits and roadmap

* Second version
2022-07-21 10:06:16 +02:00
Patrick von Platen e795a4c6f8 Fix import metadatalib 2022-07-21 04:56:46 +02:00
Patrick von Platen 4293b9f54f Release: 0.1.1 2022-07-21 04:51:37 +02:00
Patrick von Platen 0e5f2daee7 Release: 0.1.0 2022-07-21 02:35:27 +00:00
Patrick von Platen 416749ff96 modelcards and tensorboard are optional 2022-07-21 02:30:55 +00:00
Patrick von Platen b1b99b59ac some more cleaning 2022-07-21 02:11:28 +00:00
Patrick von Platen 606ac57e50 finish pndm sampler 2022-07-21 01:51:58 +00:00
Patrick von Platen 394243ce98 finish pndm sampler 2022-07-21 01:50:12 +00:00
Nathan Lambert fe98574622
fixing tests for numpy and make deterministic (ddpm) (#106)
* work in progress, fixing tests for numpy and make deterministic

* make tests pass via pytorch

* make pytorch == numpy test cleaner

* change default tensor format pndm --> pt
2022-07-21 02:24:59 +02:00
Patrick von Platen c5c9399610 correct paths for tests 2022-07-21 00:20:10 +00:00
Patrick von Platen 836f3f35c2
Rename pipelines (#115)
up
2022-07-21 01:39:46 +02:00
Patrick von Platen 9c3820d05a
Big Model Renaming (#109)
* up

* change model name

* renaming

* more changes

* up

* up

* up

* save checkpoint

* finish api / naming

* finish config renaming

* rename all weights

* finish really
2022-07-21 01:30:45 +02:00
Patrick von Platen 13e37cabe0 Merge branch 'main' of https://github.com/huggingface/diffusers into main 2022-07-20 21:02:43 +00:00
Patrick von Platen 760dcb1ffc fix score sde ve scheduler 2022-07-20 21:02:40 +00:00
Nathan Lambert 889aa6008c
PNDM API Updates, Tests Cleaning (#103)
* organize PNDM tests, begin API change

* clean timestep API PNDM

* update pipeline PNDM

* fix typo

* API clean round 2

* small nit
2022-07-20 12:47:39 -07: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-l 6b275fca49 make PIL the default output type 2022-07-20 18:28:22 +02:00
Anton Lozhkov 1b42732ced
PIL-ify the pipeline outputs (#111) 2022-07-20 18:09:51 +02:00
anton-l 9e9d2dbc59 Fix np.abs 2022-07-20 17:38:03 +02:00
Anton Lozhkov 8b4371f70f
Refactor pipeline outputs, return LDM guidance_scale (#110) 2022-07-20 17:28:06 +02:00
Patrick von Platen 919e27d357 re-add super.__init__ for all PyTorch modules 2022-07-20 13:49:00 +00:00
Sylvain Gugger ad9d252596
Add a decorator for register_to_config (#108)
* Add a decorator for register_to_config

* All models and test
2022-07-20 15:42:50 +02:00
Patrick von Platen 7e11392dfd fix ddpm scheduler 2022-07-19 23:47:04 +00:00
Patrick von Platen 1f49a343b5 hotfix 2022-07-19 23:14:03 +00:00
Patrick von Platen 936cd08488 improve loading a bit 2022-07-19 22:02:54 +00:00
Patrick von Platen 3a32b8c916 align API 2022-07-19 16:54:10 +00:00
Patrick von Platen c3a15437f8 automatic logits verification >> visual logits verification 2022-07-19 16:14:17 +00:00
Patrick von Platen 8c31925b3b
Get diffusers ready 🚀🚀🚀 (#101)
* big purge

* more fixes

* finish for now
2022-07-19 18:02:12 +02:00
Arthur 33344ed916
logits for google and compvis models (#100)
* initial commit

* quick fix
2022-07-19 18:02:04 +02:00
anton-l 7353b74ec2 Merge remote-tracking branch 'origin/main' 2022-07-19 17:12:48 +02:00
anton-l 44bb38fd8b Include model_card_template.md with the package 2022-07-19 17:07:54 +02:00
Patrick von Platen 2ea64a08ed Prepare code for big cleaning 2022-07-19 15:07:46 +00:00
Patrick von Platen 37fe8e00b2 upload 2022-07-19 15:05:40 +00:00
anton-l 0ea78f0d3b Include MANIFEST.in to package the modelcard template 2022-07-19 17:01:16 +02:00
anton-l 0e5a99bb5a Quick hacks for push_to_hub from notebooks - follow-up 2022-07-19 16:52:39 +02:00
anton-l e3c982ee29 Quick hacks for push_to_hub from notebooks 2022-07-19 16:41:13 +02:00
anton-l ab00f5d3e1 Update model names for CompVis and google 2022-07-19 15:13:22 +02:00
Patrick von Platen 3f0b44b322 improve ddpm conversion script 2022-07-19 11:24:13 +00:00
Patrick von Platen cb90fd69b4 upload code 2022-07-19 10:34:52 +00:00
Arthur f794432e81
Conversion script for ncsnpp models (#98)
* added kwargs for easier intialisation of random model

* initial commit for conversion script

* current debug script

* update

* Update

* done

* add updated debug conversion script

* style

* clean conversion script
2022-07-19 12:19:36 +02:00
Nathan Lambert 182b164f32
Fix VE SDE tests, clean API (#95)
* clean ddpm api to match ddim

* correct ve sde class

* update pipeline API for ve sde

* make style

* Apply suggestions from code review

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2022-07-19 12:12:45 +02:00
Patrick von Platen 8b42c7cecc make all tests pass 2022-07-19 00:24:10 +00:00
Patrick von Platen 66d5a1804c small fixes 2022-07-19 00:08:41 +00:00
Patrick von Platen d5acb4110a
Finalize ldm (#96)
* upload

* make checkpoint work

* finalize
2022-07-19 02:02:23 +02:00
Lysandre Debut 6cabc599a2
DDPM Conversion (#94)
* DDPM

* Fixes

* Edit tests
2022-07-19 01:59:58 +02:00
anton-l 36b459f6e6 Make tqdm calls notebook-compatible - follow-up 2022-07-18 18:43:18 +02:00