Patrick von Platen
3100bc9670
[Vae and AutoencoderKL] Final clean of LDM checkpoints ( #137 )
...
* [Vae and AutoencoderKL clean]
* save intermediate finished work
* more progress
* more progress
* finish modeling code
* save intermediate
* finish
* Correct tests
2022-07-28 10:14:34 +02:00
Anton Lozhkov
e05f03ae41
Disable test_ddpm_ddim_equality_batched until resolved ( #142 )
...
disable test_ddpm_ddim_equality_batched
2022-07-28 09:29:29 +02:00
Anton Lozhkov
6c15636b0b
Add training and batched inference test for DDPM vs DDIM ( #140 )
...
* Add torch_device to the VE pipeline
* Mark the training test with slow
2022-07-27 15:01:56 +02:00
Patrick von Platen
5311f564ed
Final fixes ( #118 )
...
final fixes before release
2022-07-21 14:36:43 +02:00
Patrick von Platen
394243ce98
finish pndm sampler
2022-07-21 01:50:12 +00:00
Nathan Lambert
fe98574622
fixing tests for numpy and make deterministic (ddpm) ( #106 )
...
* work in progress, fixing tests for numpy and make deterministic
* make tests pass via pytorch
* make pytorch == numpy test cleaner
* change default tensor format pndm --> pt
2022-07-21 02:24:59 +02:00
Patrick von Platen
c5c9399610
correct paths for tests
2022-07-21 00:20:10 +00:00
Patrick von Platen
836f3f35c2
Rename pipelines ( #115 )
...
up
2022-07-21 01:39:46 +02:00
Patrick von Platen
9c3820d05a
Big Model Renaming ( #109 )
...
* up
* change model name
* renaming
* more changes
* up
* up
* up
* save checkpoint
* finish api / naming
* finish config renaming
* rename all weights
* finish really
2022-07-21 01:30:45 +02:00
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-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
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
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
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
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
Suraj Patil
4e2674934f
add tests for 1D Up/Downsample blocks ( #72 )
2022-07-04 11:41:04 +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
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
