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

..

__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 training and batched inference test for DDPM vs DDIM (#140)

2022-07-27 15:01:56 +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