Skip to content

GitLab

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

rename image to sample

parent fe7d1363
Hide 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): ...@@ -172,7 +172,7 @@ class DDPMSchedulerTest(SchedulerCommonTest):
"beta_end": 0.02, "beta_end": 0.02,
"beta_schedule": "linear", "beta_schedule": "linear",
"variance_type": "fixed_small", "variance_type": "fixed_small",
"clip_predicted_image": True, "clip_predicted_sample": True,
} }
config.update(**kwargs) config.update(**kwargs)
...@@ -195,8 +195,8 @@ class DDPMSchedulerTest(SchedulerCommonTest): ...@@ -195,8 +195,8 @@ class DDPMSchedulerTest(SchedulerCommonTest):
self.check_over_configs(variance_type=variance) self.check_over_configs(variance_type=variance)
def test_clip_image(self): def test_clip_image(self):
for clip_predicted_image in [True, False]: for clip_predicted_sample in [True, False]:
self.check_over_configs(clip_predicted_image=clip_predicted_image) self.check_over_configs(clip_predicted_sample=clip_predicted_sample)
def test_time_indices(self): def test_time_indices(self):
for t in [0, 500, 999]: for t in [0, 500, 999]:
...@@ -251,7 +251,7 @@ class DDIMSchedulerTest(SchedulerCommonTest): ...@@ -251,7 +251,7 @@ class DDIMSchedulerTest(SchedulerCommonTest):
"beta_start": 0.0001, "beta_start": 0.0001,
"beta_end": 0.02, "beta_end": 0.02,
"beta_schedule": "linear", "beta_schedule": "linear",
"clip_predicted_image": True, "clip_predicted_sample": True,
} }
config.update(**kwargs) config.update(**kwargs)
...@@ -270,8 +270,8 @@ class DDIMSchedulerTest(SchedulerCommonTest): ...@@ -270,8 +270,8 @@ class DDIMSchedulerTest(SchedulerCommonTest):
self.check_over_configs(beta_schedule=schedule) self.check_over_configs(beta_schedule=schedule)
def test_clip_image(self): def test_clip_image(self):
for clip_predicted_image in [True, False]: for clip_predicted_sample in [True, False]:
self.check_over_configs(clip_predicted_image=clip_predicted_image) self.check_over_configs(clip_predicted_sample=clip_predicted_sample)
def test_time_indices(self): def test_time_indices(self):
for t in [1, 10, 49]: 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