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
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
8b42c7cecc
make all tests pass
2022-07-19 00:24:10 +00:00
66d5a1804c
small fixes
2022-07-19 00:08:41 +00:00
d5acb4110a
Finalize ldm ( #96 )
...
* upload
* make checkpoint work
* finalize
2022-07-19 02:02:23 +02:00
6cabc599a2
DDPM Conversion ( #94 )
...
* DDPM
* Fixes
* Edit tests
2022-07-19 01:59:58 +02:00
36b459f6e6
Make tqdm calls notebook-compatible - follow-up
2022-07-18 18:43:18 +02:00
1820024005
Make tqdm calls notebook-compatible
2022-07-18 18:39:39 +02:00
ffe7b93b60
DDIM resolution->image_size
2022-07-18 12:23:27 +02:00
f82ebb9a03
fix some model tests
2022-07-18 01:29:40 +00:00
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
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
b5c684f042
fix flaky cpu test
2022-07-15 19:49:05 +00:00
da8e87e201
use real checkpoint
2022-07-15 19:13:39 +00:00
43bbc78123
adapt test
2022-07-15 18:37:15 +00:00
1c14ce9509
fix local subfolder
2022-07-15 17:55:20 +00:00
29628acbec
renaming of api
2022-07-15 17:29:14 +00:00
9d2fc6b535
some fixes
2022-07-15 17:22:28 +00:00
3f1e95928e
Fix conversion script
2022-07-15 17:00:41 +00:00
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
e5f3415fbd
Update README.md
2022-07-15 17:28:04 +02:00
f5ca5af6ce
add to readme
2022-07-15 14:06:45 +00:00
2ac19ff190
Merge branch 'main' of https://github.com/huggingface/diffusers into main
2022-07-15 14:03:11 +00:00
badc5517ff
fix small bug
2022-07-15 14:03:08 +00:00
a8fc1560c6
Update README.md
2022-07-15 15:06:38 +02:00
f448360bd0
Finish scheduler API ( #91 )
...
* finish
* up
2022-07-15 15:04:01 +02:00
97e1e3ba76
finalize model API
2022-07-15 10:48:30 +00:00
dacabaa47f
readme vp typo fix
2022-07-14 14:39:19 -07:00
6d5ef87e6b
[DDPM] Make DDPM work ( #88 )
...
* up
* finish
* uP
2022-07-14 19:46:04 +02:00
e7fe901e5e
save intermediate ( #87 )
...
* save intermediate
* up
* up
2022-07-14 12:29:06 +02:00
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
2a69c0b7b8
The big purge -> remove everything except vision for now
2022-07-13 11:42:40 +00:00
c8c0c0e846
quick fix
2022-07-13 10:28:46 +00:00
5e12d5c691
Clean uncond unet more ( #85 )
...
* up
* finished clean up
* remove @
2022-07-13 12:21:11 +02:00
8aed37c1bd
some more refactor
2022-07-12 19:35:47 +00:00
06c79730d0
Add unconditional image generation ( #79 )
...
* uP
* finish downsampling layers
* finish major refactor
* remove bugus file
2022-07-12 18:34:41 +02:00
ea8d58ea91
[MidBlock] Fix mid block ( #78 )
...
* upload files
* finish
2022-07-05 15:05:41 +02:00
c352faeae3
Add MidBlock to Grad-TTS ( #74 )
...
Finish
2022-07-04 15:06:00 +02:00
107986639d
Fix attention for Glide ( #75 )
2022-07-04 14:55:56 +02:00
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
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
94566e6dd8
update mid block ( #70 )
...
* update mid block
* finish mid block
2022-07-04 11:52:22 +02:00
4e2674934f
add tests for 1D Up/Downsample blocks ( #72 )
2022-07-04 11:41:04 +02:00
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
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
c524244f49
[Resnet] Remove unnecessary functions / classes ( #67 )
...
Remove unnecessary functions / classes
2022-07-03 19:17:25 +02:00
d224c6373f
Resnet => Resnet2D ( #66 )
2022-07-03 18:58:58 +02:00
44705a648b
[ResNet] Refactor resnet from VAE ( #65 )
2022-07-03 18:43:43 +02:00
a7b0047e0f
some clean up
2022-07-01 18:14:46 +00:00
dcb9070bc2
quick fix to include non-fir kernels for sde-vp
2022-07-01 17:56:59 +00:00
Arthur