Commit Graph

231 Commits

Author SHA1 Message Date
Nathan Lambert 3a5c87055c
add RL test, remove conds from RL model input 2022-06-27 14:48:15 -04:00
Patrick von Platen a2b72faff7 Merge branch 'main' of https://github.com/huggingface/diffusers into main 2022-06-27 17:20:20 +00:00
Patrick von Platen c9504bba10 add tests for sde ve vp models 2022-06-27 17:20:15 +00:00
patil-suraj 26ea58d4e1 Merge branch 'main' of https://github.com/huggingface/diffusers into main 2022-06-27 18:04:49 +02:00
patil-suraj 7b9b946cb2 add tests for downsample block 2022-06-27 18:03:51 +02: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
patil-suraj dc7c49e4e4 add tests for upsample blocks 2022-06-27 15:50:54 +02:00
Patrick von Platen 3986741b8b add another ldm fast test 2022-06-27 13:25:26 +00:00
Patrick von Platen 6846ee2ac4 finalize position embeddings 2022-06-27 11:43:08 +00:00
Patrick von Platen c7a39d38ad refactor all sinus embeddings 2022-06-27 11:37:37 +00:00
Patrick von Platen 02a76c2c81 consolidate timestep embeds 2022-06-27 10:14:54 +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
Patrick von Platen 45a09bebf3 add first files 2022-06-27 10:46:39 +02:00
Patrick von Platen ba264419f4 finish vp 2022-06-27 00:07:57 +00:00
Patrick von Platen dc6d028654 add vp sampler 2022-06-26 23:41:55 +00:00
Patrick von Platen 433cb3f801 clean up sde ve more 2022-06-25 18:25:43 +00:00
Patrick von Platen f941fc9917 refactor tts sampler a bit 2022-06-22 23:15:57 +02:00
Patrick von Platen 0244e2af4c correct diffusion test 2022-06-22 18:41:14 +02:00
Patrick von Platen 48269070d2 more fixes 2022-06-22 13:40:08 +00:00
Patrick von Platen d0032c6095 refactor naming 2022-06-22 12:38:36 +00:00
patil-suraj e3bf932404 don't hardcode device in tests 2022-06-21 12:02:21 +02:00
patil-suraj ac00dad756 add GLIDETextToImageUNetModelTests 2022-06-21 12:01:07 +02:00
anton-l 9c82c32ba7 make style 2022-06-21 10:43:40 +02:00
patil-suraj 80898b5234 add UNetGradTTSModelTests 2022-06-20 15:57:58 +02:00
patil-suraj e5675fad5d remove prints from tests 2022-06-20 14:47:13 +02:00
patil-suraj 95a45f5b3a add UNetLDMModelTests 2022-06-20 14:45:58 +02:00
patil-suraj 646e16fe06 fix test_output_pretrained for GLIDESuperResUNetModel 2022-06-20 14:27:37 +02:00
Patrick von Platen 08c852290a add license disclaimers to schedulers 2022-06-20 13:06:31 +02:00
patil-suraj e007c797b1 add GLIDESuperResUNetModel 2022-06-17 19:04:07 +02:00
patil-suraj 12da0fe10d Merge branch 'main' into model-tests 2022-06-17 16:37:45 +02:00
patil-suraj cf6cd39572 finish tests for UNet 2022-06-17 16:36:51 +02:00
Patrick von Platen 1997b90838 image->sample in schedule tests 2022-06-17 15:51:33 +02:00
Patrick von Platen b2274ece73 finish pndm scheduler 2022-06-17 15:51:03 +02:00
patil-suraj 7dc71897b3 add UnetModelTests 2022-06-17 13:49:26 +02:00
patil-suraj d76bc43720 add skeleton for model tests 2022-06-17 13:36:59 +02:00
Patrick von Platen 8c1f51978c make clip name shorter 2022-06-17 12:11:40 +02:00
Patrick von Platen 13a78b3cd3 rename image to sample 2022-06-17 12:09:13 +02:00
Patrick von Platen fe7d136324 correct dict 2022-06-17 11:55:02 +02:00
Patrick von Platen 5e6f500038 rename register to register_to_config 2022-06-17 10:58: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
patil-suraj 542c78686f Merge branch 'main' of https://github.com/huggingface/diffusers into main 2022-06-14 12:51:40 +02:00
patil-suraj 147d8e0702 add test for loading model from pipeline module 2022-06-14 12:50:40 +02:00
anton-l 57243fd565 GLIDE integration test 2022-06-14 12:37:28 +02:00
anton-l bb30664285 Move the training example 2022-06-14 11:33:24 +02:00
Patrick von Platen 809591b7b6 improve pndm 2022-06-13 16:29:22 +00:00
Patrick von Platen 27266abc9f rename schedulers 2022-06-13 10:39:53 +02:00
Patrick von Platen 12b10cbe09 finish refactor 2022-06-12 21:20:39 +00:00
Patrick von Platen 2d97544dc7 add more tests schedulers 2022-06-12 19:56:13 +00:00
Patrick von Platen e83ff11f57 make tests pass 2022-06-12 17:59:39 +00:00
Patrick von Platen a020285e8e update 2022-06-12 19:12:01 +02:00
Patrick von Platen e01bcbb765 rename to step 2022-06-12 19:07:56 +02:00
patil-suraj 96306533cb add test for ldm 2022-06-10 18:37:45 +02:00
Patrick von Platen 9d32a26579 save intermediate 2022-06-10 13:12:23 +02:00
Patrick von Platen cbb19ee84e fix setup 2022-06-09 14:06:58 +02:00
Patrick von Platen 2234877e01 fix tests 2022-06-09 11:02:32 +00:00
Patrick von Platen b02d0d6be3 merge 2022-06-09 12:39:31 +02:00
Patrick von Platen 49257b4abf finish transformers removal 2022-06-09 12:36:37 +02:00
Patrick von Platen 7a1323b62f add first version of ddim 2022-06-08 09:42:31 +00:00
Patrick von Platen 3e801673d6 remove ipdb 2022-06-07 17:00:21 +00:00
Patrick von Platen f8a9bb6f63 merge 2022-06-07 16:59:48 +00:00
Patrick von Platen 6f88cc92e9 adapt final unpreciseness 2022-06-07 16:59:12 +00:00
Patrick von Platen 89af440e32 check with other device 2022-06-07 18:55:10 +02:00
Patrick von Platen b76eea0412 check with other device 2022-06-07 18:53:10 +02:00
Patrick von Platen 5da71f8fa3 fix generator 2 2022-06-07 16:22:12 +00:00
Patrick von Platen 46d20d2d76 fix random seed 2022-06-07 18:20:14 +02:00
patil-suraj f39020bd8a clip => clipped 2022-06-07 16:34:44 +02:00
patil-suraj 733546210e fix tests 2022-06-07 15:43:08 +02:00
patil-suraj d8287fcd1d fix issues with loading, add test for pipeline 2022-06-07 15:40:36 +02:00
Patrick von Platen ef4365c6ef up 2022-06-07 13:03:53 +00:00
Patrick von Platen fe3137304b improve 2022-06-06 17:03:41 +02:00
Patrick von Platen 8cb5e69415 add pretrained model and pretrained sampler 2022-06-02 00:25:48 +02:00
Patrick von Platen 18ef809c4d add another test 2022-05-31 14:46:20 +02:00
Patrick von Platen e779b250e1 add first template for DDPM forward 2022-05-31 14:27:59 +02:00