From 85f9d92883d5a948e3cecb0e62a80780d897e3e1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?M=2E=20Tolga=20Cang=C3=B6z?= <46008593+standardAI@users.noreply.github.com> Date: Tue, 19 Mar 2024 05:58:47 +0300 Subject: [PATCH] Fix conditional statement in test_schedulers.py (#7323) Co-authored-by: Sayak Paul --- tests/schedulers/test_schedulers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/schedulers/test_schedulers.py b/tests/schedulers/test_schedulers.py index 58acf42576..fc7f22d2a8 100755 --- a/tests/schedulers/test_schedulers.py +++ b/tests/schedulers/test_schedulers.py @@ -716,7 +716,7 @@ class SchedulerCommonTest(unittest.TestCase): # Get valid timestep based on sigma_max, which should always be in timestep schedule. scaled_sigma_max = scheduler.sigma_to_t(scheduler.config.sigma_max) scaled_sample = scheduler.scale_model_input(sample, scaled_sigma_max) - if scheduler_class == EDMEulerScheduler: + elif scheduler_class == EDMEulerScheduler: scaled_sample = scheduler.scale_model_input(sample, scheduler.timesteps[-1]) else: scaled_sample = scheduler.scale_model_input(sample, 0.0)