Fix conditional statement in test_schedulers.py (#7323)

Co-authored-by: Sayak Paul <[email protected]>

Author: tolgacangoz

tests/schedulers/test_schedulers.py

@@ -716,7 +716,7 @@ def test_add_noise_device(self):
                 # 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)