AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

rename image to sample

Browse Source
This commit is contained in:
Patrick von Platen
2022-06-17 12:09:13 +02:00
parent fe7d136324
commit 13a78b3cd3
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_scheduler.py
View File

@@ -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]: