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