diffusers/tests
Patrick von Platen 9e1b1ca49d
[Tests] Make sure tests are on GPU (#269)
* [Tests] Make sure tests are on GPU

* move more models

* speed up tests
2022-08-29 15:58:11 +02:00
..
__init__.py fix issues with loading, add test for pipeline 2022-06-07 15:40:36 +02:00
test_config.py [Tests] Make sure tests are on GPU (#269) 2022-08-29 15:58:11 +02:00
test_layers_utils.py Revive `make quality` (#203) 2022-08-17 15:22:04 +02:00
test_modeling_common.py split tests_modeling_utils (#223) 2022-08-24 13:27:16 +02:00
test_models_unet.py [Tests] Make sure tests are on GPU (#269) 2022-08-29 15:58:11 +02:00
test_models_vae.py [Tests] Make sure tests are on GPU (#269) 2022-08-29 15:58:11 +02:00
test_models_vq.py [Tests] Make sure tests are on GPU (#269) 2022-08-29 15:58:11 +02:00
test_pipelines.py [Tests] Make sure tests are on GPU (#269) 2022-08-29 15:58:11 +02:00
test_scheduler.py added test workflow and fixed failing test (#237) 2022-08-24 13:46:53 +02:00
test_training.py Add training and batched inference test for DDPM vs DDIM (#140) 2022-07-27 15:01:56 +02:00