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
anton-l
1820024005
Make tqdm calls notebook-compatible
2022-07-18 18:39:39 +02:00
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
