Skip to content

GitLab

Commit 13a78b3c authored by Patrick von Platen's avatar Patrick von Platen
Browse files

rename image to sample

parent fe7d1363
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
tests/test_scheduler.py
View file @ 13a78b3c
......@@ -172,7 +172,7 @@ class DDPMSchedulerTest(SchedulerCommonTest):
"beta_end": 0.02,
"beta_schedule": "linear",
"variance_type": "fixed_small",
"clip_predicted_image": True,
"clip_predicted_sample": True,
}
config.update(**kwargs)
......@@ -195,8 +195,8 @@ class DDPMSchedulerTest(SchedulerCommonTest):
self.check_over_configs(variance_type=variance)
def test_clip_image(self):
for clip_predicted_image in [True, False]:
self.check_over_configs(clip_predicted_image=clip_predicted_image)
for clip_predicted_sample in [True, False]:
self.check_over_configs(clip_predicted_sample=clip_predicted_sample)
def test_time_indices(self):
for t in [0, 500, 999]:
......@@ -251,7 +251,7 @@ class DDIMSchedulerTest(SchedulerCommonTest):
"beta_start": 0.0001,
"beta_end": 0.02,
"beta_schedule": "linear",
"clip_predicted_image": True,
"clip_predicted_sample": True,
}
config.update(**kwargs)
......@@ -270,8 +270,8 @@ class DDIMSchedulerTest(SchedulerCommonTest):
self.check_over_configs(beta_schedule=schedule)
def test_clip_image(self):
for clip_predicted_image in [True, False]:
self.check_over_configs(clip_predicted_image=clip_predicted_image)
for clip_predicted_sample in [True, False]:
self.check_over_configs(clip_predicted_sample=clip_predicted_sample)
def test_time_indices(self):
for t in [1, 10, 49]:
......
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