Commit Graph

105 Commits

Author SHA1 Message Date
Patrick von Platen 3a32b8c916 align API 2022-07-19 16:54:10 +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
Patrick von Platen 37fe8e00b2 upload 2022-07-19 15:05:40 +00: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
Patrick von Platen f82ebb9a03 fix some model tests 2022-07-18 01:29:40 +00:00
Nathan Lambert 63c68d979a
VE/VP SDE updates (#90)
* improve comments for sde_ve scheduler, init tests

* more comments, tweaking pipelines

* timesteps --> num_training_timesteps, some comments

* merge cpu test, add m1 data

* fix scheduler tests with num_train_timesteps

* make np compatible, add tests for sde ve

* minor default variable fixes

* make style and fix-copies

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2022-07-18 03:08:08 +02:00
Patrick von Platen ba3c9a9a3a
[SDE] Merge to unconditional model (#89)
* up

* more

* uP

* make dummy test pass

* save intermediate

* p

* p

* finish

* finish

* finish
2022-07-18 02:52:37 +02:00
Patrick von Platen b5c684f042 fix flaky cpu test 2022-07-15 19:49:05 +00:00
Patrick von Platen da8e87e201 use real checkpoint 2022-07-15 19:13:39 +00:00
Patrick von Platen 43bbc78123 adapt test 2022-07-15 18:37:15 +00:00
Patrick von Platen 9d2fc6b535 some fixes 2022-07-15 17:22:28 +00:00
Patrick von Platen f448360bd0
Finish scheduler API (#91)
* finish

* up
2022-07-15 15:04:01 +02:00
Patrick von Platen 97e1e3ba76 finalize model API 2022-07-15 10:48:30 +00:00
Patrick von Platen 6d5ef87e6b
[DDPM] Make DDPM work (#88)
* up

* finish

* uP
2022-07-14 19:46:04 +02:00
Patrick von Platen e7fe901e5e
save intermediate (#87)
* save intermediate

* up

* up
2022-07-14 12:29:06 +02:00
Patrick von Platen 2a69c0b7b8 The big purge -> remove everything except vision for now 2022-07-13 11:42:40 +00:00
Patrick von Platen c8c0c0e846 quick fix 2022-07-13 10:28:46 +00:00
Patrick von Platen 5e12d5c691
Clean uncond unet more (#85)
* up

* finished clean up

* remove @
2022-07-13 12:21:11 +02:00
Patrick von Platen 8aed37c1bd some more refactor 2022-07-12 19:35:47 +00:00
Patrick von Platen 06c79730d0
Add unconditional image generation (#79)
* uP

* finish downsampling layers

* finish major refactor

* remove bugus file
2022-07-12 18:34:41 +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
patil-suraj f26d3011c7 fix ldm uncond pipeline 2022-07-01 17:19:26 +02:00
Patrick von Platen db7ec72dd8 up 2022-06-30 22:29:18 +00:00
Patrick von Platen efe1e60e12 merge glide into resnets 2022-06-30 22:24:22 +00:00
Patrick von Platen db934c6750 fix more tests 2022-06-30 21:47:40 +00:00
Patrick von Platen c1c4dea98d correct tests ncsnpp 2022-06-30 15:54:00 +00:00
Patrick von Platen 3dbd6a8f4d up 2022-06-30 14:54:31 +00:00
patil-suraj 859ffea2b1 add test for ldm uncond 2022-06-29 15:25:51 +02:00
patil-suraj 333a8da678 add tests for AutoencoderKL 2022-06-29 13:52:04 +02:00
patil-suraj bae04ea9d8 add test for VQModel 2022-06-29 12:34:24 +02:00
Patrick von Platen 31d1f3c8c0 final fix 2022-06-28 22:59:21 +00:00
Patrick von Platen 79db3eb6ca fix tests 2022-06-28 17:36:56 +00:00
Patrick von Platen c45fd7498c merge unet attention into glide attention 2022-06-28 17:31:44 +00:00
Patrick von Platen bdecc3cffd move pipelines into folders 2022-06-28 10:47:47 +00:00
Patrick von Platen a859b1992b fix rl model tests 2022-06-28 09:50:21 +00:00
Nathan Lambert 3a5c87055c
add RL test, remove conds from RL model input 2022-06-27 14:48:15 -04:00
Patrick von Platen c9504bba10 add tests for sde ve vp models 2022-06-27 17:20:15 +00:00
Patrick von Platen 4261c3aadf Make style 2022-06-27 15:59:04 +00:00
Patrick von Platen 932ce05d97 cancel einops 2022-06-27 15:39:41 +00:00
patil-suraj abcb25978a Merge branch 'main' of https://github.com/huggingface/diffusers into main 2022-06-27 16:25:52 +02:00
patil-suraj 183056f243 consolidate Upsample 2022-06-27 16:25:47 +02:00
Patrick von Platen 3986741b8b add another ldm fast test 2022-06-27 13:25:26 +00:00
patil-suraj 9b9afc9726 actually fix test_ldm_text2img_fast 2022-06-27 11:46:50 +02:00
patil-suraj b7f0ce5b39 fix test_ldm_text2img_fast 2022-06-27 11:44:05 +02:00
patil-suraj 6921393ae2 add fast test for ldm 2022-06-27 11:42:52 +02:00
patil-suraj 17bf65e186 skip test_ldm_text2img for now 2022-06-27 11:39:19 +02:00
Patrick von Platen 1a0331a78a fix some tests on gpu 2022-06-27 09:07:57 +00:00