diffusers/tests
Anton Lozhkov 66fd3ec70d
[CI] try to fix GPU OOMs between tests and excessive tqdm logging (#323)
* Fix tqdm and OOM

* tqdm auto

* tqdm is still spamming try to disable it altogether

* rather just set the pipe config, to keep the global tqdm clean

* style
2022-09-02 13:18:49 +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 [CI] try to fix GPU OOMs between tests and excessive tqdm logging (#323) 2022-09-02 13:18:49 +02:00
test_scheduler.py [PNDM Scheduler] format timesteps attrs to np arrays (#273) 2022-08-31 14:12:08 +02:00
test_training.py Fix nondeterministic tests for GPU runs (#314) 2022-09-01 15:25:39 +02:00