Skip to content

GitLab

Unverified Commit 85f9d928 authored by M. Tolga Cangöz's avatar M. Tolga Cangöz Committed by GitHub
Browse files

Fix conditional statement in test_schedulers.py (#7323) 


Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent 916d9812
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/schedulers/test_schedulers.py
View file @ 85f9d928
...@@ -716,7 +716,7 @@ class SchedulerCommonTest(unittest.TestCase): ...@@ -716,7 +716,7 @@ class SchedulerCommonTest(unittest.TestCase):
# Get valid timestep based on sigma_max, which should always be in timestep schedule. # 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_sigma_max = scheduler.sigma_to_t(scheduler.config.sigma_max)
scaled_sample = scheduler.scale_model_input(sample, scaled_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]) scaled_sample = scheduler.scale_model_input(sample, scheduler.timesteps[-1])
else: else:
scaled_sample = scheduler.scale_model_input(sample, 0.0) scaled_sample = scheduler.scale_model_input(sample, 0.0)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment