Anton Lozhkov dd10da76a7 Add an alternative Karras et al. stochastic scheduler for VE models (#160) ...

* karras + VE, not flexible yet

* Fix inputs incompatibility with the original unet

* Roll back sigma scaling

* Apply suggestions from code review

* Old comment

* Fix doc

2022-08-09 15:58:30 +02:00

..

__init__.py

fix issues with loading, add test for pipeline

2022-06-07 15:40:36 +02:00

test_layers_utils.py

add tests for 1D Up/Downsample blocks (#72)

2022-07-04 11:41:04 +02:00

test_modeling_utils.py

Add an alternative Karras et al. stochastic scheduler for VE models (#160)

2022-08-09 15:58:30 +02:00

test_scheduler.py

finish pndm sampler

2022-07-21 01:50:12 +00:00

test_training.py

Add training and batched inference test for DDPM vs DDIM (#140)

2022-07-27 15:01:56 +02:00