Commit Graph

801 Commits

Author SHA1 Message Date
anton-l ffe7b93b60 DDIM resolution->image_size 2022-07-18 12:23:27 +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 1c14ce9509 fix local subfolder 2022-07-15 17:55:20 +00:00
Patrick von Platen 29628acbec renaming of api 2022-07-15 17:29:14 +00:00
Patrick von Platen 9d2fc6b535 some fixes 2022-07-15 17:22:28 +00:00
Patrick von Platen 3f1e95928e Fix conversion script 2022-07-15 17:00:41 +00:00
Lysandre Debut 87060e6a9c
LDM conversion script (#92)
Conversion script

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2022-07-15 17:29:34 +02:00
Patrick von Platen e5f3415fbd
Update README.md 2022-07-15 17:28:04 +02:00
Patrick von Platen f5ca5af6ce add to readme 2022-07-15 14:06:45 +00:00
Patrick von Platen 2ac19ff190 Merge branch 'main' of https://github.com/huggingface/diffusers into main 2022-07-15 14:03:11 +00:00
Patrick von Platen badc5517ff fix small bug 2022-07-15 14:03:08 +00:00
Patrick von Platen a8fc1560c6
Update README.md 2022-07-15 15:06:38 +02: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
Nathan Lambert dacabaa47f
readme vp typo fix 2022-07-14 14:39:19 -07: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
Nathan Lambert c3d78cd306
Docs (#45)
* first pass at docs structure

* minor reformatting, add github actions for docs

* populate docs (primarily from README, some writing)
2022-07-13 08:42:05 -07: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
Patrick von Platen ea8d58ea91
[MidBlock] Fix mid block (#78)
* upload files

* finish
2022-07-05 15:05:41 +02:00
Patrick von Platen c352faeae3
Add MidBlock to Grad-TTS (#74)
Finish
2022-07-04 15:06:00 +02:00
Anton Lozhkov 107986639d
Fix attention for Glide (#75) 2022-07-04 14:55:56 +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
Patrick von Platen 94566e6dd8
update mid block (#70)
* update mid block

* finish mid block
2022-07-04 11:52:22 +02:00
Suraj Patil 4e2674934f
add tests for 1D Up/Downsample blocks (#72) 2022-07-04 11:41:04 +02:00
Suraj Patil 53a42d0a0c
Simplify FirUp/down, unet sde (#71)
* refactor fir up/down sample

* remove variance scaling

* remove variance scaling from unet sde

* refactor Linear

* style

* actually remove variance scaling

* add back upsample_2d, downsample_2d

* style

* fix FirUpsample2D
2022-07-04 11:23:19 +02:00
Patrick von Platen 321f9791d6
Downsample / Upsample - clean to 1D and 2D (#68)
* make unet rl work

* uploaad files / code

* upload files

* make style correct

* finish
2022-07-03 22:26:33 +02:00
Patrick von Platen c524244f49
[Resnet] Remove unnecessary functions / classes (#67)
Remove unnecessary functions / classes
2022-07-03 19:17:25 +02:00
Patrick von Platen d224c6373f
Resnet => Resnet2D (#66) 2022-07-03 18:58:58 +02:00
Patrick von Platen 44705a648b
[ResNet] Refactor resnet from VAE (#65) 2022-07-03 18:43:43 +02:00
Patrick von Platen a7b0047e0f some clean up 2022-07-01 18:14:46 +00:00
Patrick von Platen dcb9070bc2 quick fix to include non-fir kernels for sde-vp 2022-07-01 17:56:59 +00:00
Patrick von Platen 11667d08d3
Merge pull request #59 from huggingface/fuse_final_resnets
[Resnet] Merge final 2D resnet
2022-07-01 19:32:36 +02:00
Patrick von Platen 221de0edee correct 2022-07-01 17:28:29 +00:00
Patrick von Platen 0eac7bd682 small fix 2022-07-01 17:20:30 +00:00
Patrick von Platen 1e7e23a9c6 Merge branch 'fuse_final_resnets' of https://github.com/huggingface/diffusers into fuse_final_resnets 2022-07-01 16:42:26 +00:00
Patrick von Platen b8415bb480 remove bogus files 2022-07-01 16:42:24 +00:00
Patrick von Platen 3a15afacab delete bogus files 2022-07-01 16:20:46 +00:00
Patrick von Platen 571e4062e5 merge from master 2022-07-01 16:20:05 +00:00
Patrick von Platen 14bd3567b0 update 2022-07-01 15:45:40 +00:00