Fix conditional statement in test_schedulers.py (#7323)

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
2024-03-19 05:58:47 +03:00
parent 916d9812a8
commit 85f9d92883
1 changed files with 1 additions and 1 deletions
@@ -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)