From ad6cd712ab0919ebb264fab8729a4ad9288633b4 Mon Sep 17 00:00:00 2001 From: enzymezoo-code Date: Tue, 5 Sep 2023 15:45:35 -0500 Subject: [PATCH] Loop all samplers/schedulers in test_inference.py --- tests/inference/test_inference.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/inference/test_inference.py b/tests/inference/test_inference.py index 28cfa601e..66dee6f55 100644 --- a/tests/inference/test_inference.py +++ b/tests/inference/test_inference.py @@ -141,8 +141,8 @@ SCHEDULERS = ["normal", "karras", "exponential", "sgm_uniform", "simple", "ddim_ SAMPLERS = ["euler", "euler_ancestral", "heun", "dpm_2", "dpm_2_ancestral", "lms", "dpm_fast", "dpm_adaptive", "dpmpp_2s_ancestral", "dpmpp_sde", "dpmpp_sde_gpu", "dpmpp_2m", "dpmpp_2m_sde", "dpmpp_2m_sde_gpu", "dpmpp_3m_sde", "dpmpp_3m_sde_gpu", "ddim", "uni_pc", "uni_pc_bh2"] -sampler_list = [SAMPLERS[0]] -scheduler_list = [SCHEDULERS[0]] +sampler_list = SAMPLERS +scheduler_list = SCHEDULERS @pytest.mark.inference @pytest.mark.parametrize("sampler", sampler_list) @pytest.mark.parametrize("scheduler", scheduler_list)