Skip to content

Commit 64e48b7

Browse files
fboulnoisfboulnois
committed
test: update tests for core pipelines
1 parent e941e29 commit 64e48b7

8 files changed

+41
-41
lines changed

‎tests/pipelines/altdiffusion/test_alt_diffusion_img2img.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ def test_stable_diffusion_img2img_default_case(self):
141141
guidance_scale=6.0,
142142
num_inference_steps=2,
143143
output_type="np",
144-
init_image=init_image,
144+
image=init_image,
145145
)
146146

147147
image = output.images
@@ -153,7 +153,7 @@ def test_stable_diffusion_img2img_default_case(self):
153153
guidance_scale=6.0,
154154
num_inference_steps=2,
155155
output_type="np",
156-
init_image=init_image,
156+
image=init_image,
157157
return_dict=False,
158158
)[0]
159159

@@ -204,7 +204,7 @@ def test_stable_diffusion_img2img_fp16(self):
204204
generator=generator,
205205
num_inference_steps=2,
206206
output_type="np",
207-
init_image=init_image,
207+
image=init_image,
208208
).images
209209

210210
assert image.shape == (1, 32, 32, 3)
@@ -243,7 +243,7 @@ def test_stable_diffusion_img2img_pipeline_default(self):
243243
generator = torch.Generator(device=torch_device).manual_seed(0)
244244
output = pipe(
245245
prompt=prompt,
246-
init_image=init_image,
246+
image=init_image,
247247
strength=0.75,
248248
guidance_scale=7.5,
249249
generator=generator,

‎tests/pipelines/latent_diffusion/test_latent_diffusion_superresolution.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def test_inference_superresolution(self):
7979
init_image = self.dummy_image.to(device)
8080

8181
generator = torch.Generator(device=device).manual_seed(0)
82-
image = ldm(init_image, generator=generator, num_inference_steps=2, output_type="numpy").images
82+
image = ldm(image=init_image, generator=generator, num_inference_steps=2, output_type="numpy").images
8383

8484
image_slice = image[0, -3:, -3:, -1]
8585

@@ -103,7 +103,7 @@ def test_inference_superresolution(self):
103103
ldm.set_progress_bar_config(disable=None)
104104

105105
generator = torch.Generator(device=torch_device).manual_seed(0)
106-
image = ldm(init_image, generator=generator, num_inference_steps=20, output_type="numpy").images
106+
image = ldm(image=init_image, generator=generator, num_inference_steps=20, output_type="numpy").images
107107

108108
image_slice = image[0, -3:, -3:, -1]
109109

‎tests/pipelines/stable_diffusion/test_cycle_diffusion.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,7 @@ def test_stable_diffusion_cycle(self):
186186
source_prompt=source_prompt,
187187
generator=generator,
188188
num_inference_steps=2,
189-
init_image=init_image,
189+
image=init_image,
190190
eta=0.1,
191191
strength=0.8,
192192
guidance_scale=3,
@@ -244,7 +244,7 @@ def test_stable_diffusion_cycle_fp16(self):
244244
source_prompt=source_prompt,
245245
generator=generator,
246246
num_inference_steps=2,
247-
init_image=init_image,
247+
image=init_image,
248248
eta=0.1,
249249
strength=0.8,
250250
guidance_scale=3,
@@ -297,7 +297,7 @@ def test_cycle_diffusion_pipeline_fp16(self):
297297
output = pipe(
298298
prompt=prompt,
299299
source_prompt=source_prompt,
300-
init_image=init_image,
300+
image=init_image,
301301
num_inference_steps=100,
302302
eta=0.1,
303303
strength=0.85,
@@ -336,7 +336,7 @@ def test_cycle_diffusion_pipeline(self):
336336
output = pipe(
337337
prompt=prompt,
338338
source_prompt=source_prompt,
339-
init_image=init_image,
339+
image=init_image,
340340
num_inference_steps=100,
341341
eta=0.1,
342342
strength=0.85,

‎tests/pipelines/stable_diffusion/test_onnx_stable_diffusion_img2img.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ def test_inference_default_pndm(self):
7272
generator = np.random.RandomState(0)
7373
output = pipe(
7474
prompt=prompt,
75-
init_image=init_image,
75+
image=init_image,
7676
strength=0.75,
7777
guidance_scale=7.5,
7878
num_inference_steps=10,
@@ -110,7 +110,7 @@ def test_inference_k_lms(self):
110110
generator = np.random.RandomState(0)
111111
output = pipe(
112112
prompt=prompt,
113-
init_image=init_image,
113+
image=init_image,
114114
strength=0.75,
115115
guidance_scale=7.5,
116116
num_inference_steps=10,

‎tests/pipelines/stable_diffusion/test_onnx_stable_diffusion_inpaint_legacy.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ def test_inference(self):
8080
generator = np.random.RandomState(0)
8181
output = pipe(
8282
prompt=prompt,
83-
init_image=init_image,
83+
image=init_image,
8484
mask_image=mask_image,
8585
strength=0.75,
8686
guidance_scale=7.5,

‎tests/pipelines/stable_diffusion/test_stable_diffusion_img2img.py‎

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -188,7 +188,7 @@ def test_stable_diffusion_img2img_default_case(self):
188188
guidance_scale=6.0,
189189
num_inference_steps=2,
190190
output_type="np",
191-
init_image=init_image,
191+
image=init_image,
192192
)
193193

194194
image = output.images
@@ -200,7 +200,7 @@ def test_stable_diffusion_img2img_default_case(self):
200200
guidance_scale=6.0,
201201
num_inference_steps=2,
202202
output_type="np",
203-
init_image=init_image,
203+
image=init_image,
204204
return_dict=False,
205205
)[0]
206206

@@ -245,7 +245,7 @@ def test_stable_diffusion_img2img_negative_prompt(self):
245245
guidance_scale=6.0,
246246
num_inference_steps=2,
247247
output_type="np",
248-
init_image=init_image,
248+
image=init_image,
249249
)
250250
image = output.images
251251
image_slice = image[0, -3:, -3:, -1]
@@ -285,7 +285,7 @@ def test_stable_diffusion_img2img_multiple_init_images(self):
285285
guidance_scale=6.0,
286286
num_inference_steps=2,
287287
output_type="np",
288-
init_image=init_image,
288+
image=init_image,
289289
)
290290

291291
image = output.images
@@ -328,7 +328,7 @@ def test_stable_diffusion_img2img_k_lms(self):
328328
guidance_scale=6.0,
329329
num_inference_steps=2,
330330
output_type="np",
331-
init_image=init_image,
331+
image=init_image,
332332
)
333333
image = output.images
334334

@@ -339,7 +339,7 @@ def test_stable_diffusion_img2img_k_lms(self):
339339
guidance_scale=6.0,
340340
num_inference_steps=2,
341341
output_type="np",
342-
init_image=init_image,
342+
image=init_image,
343343
return_dict=False,
344344
)
345345
image_from_tuple = output[0]
@@ -382,7 +382,7 @@ def test_stable_diffusion_img2img_num_images_per_prompt(self):
382382
prompt,
383383
num_inference_steps=2,
384384
output_type="np",
385-
init_image=init_image,
385+
image=init_image,
386386
).images
387387

388388
assert images.shape == (1, 32, 32, 3)
@@ -393,7 +393,7 @@ def test_stable_diffusion_img2img_num_images_per_prompt(self):
393393
[prompt] * batch_size,
394394
num_inference_steps=2,
395395
output_type="np",
396-
init_image=init_image,
396+
image=init_image,
397397
).images
398398

399399
assert images.shape == (batch_size, 32, 32, 3)
@@ -404,7 +404,7 @@ def test_stable_diffusion_img2img_num_images_per_prompt(self):
404404
prompt,
405405
num_inference_steps=2,
406406
output_type="np",
407-
init_image=init_image,
407+
image=init_image,
408408
num_images_per_prompt=num_images_per_prompt,
409409
).images
410410

@@ -416,7 +416,7 @@ def test_stable_diffusion_img2img_num_images_per_prompt(self):
416416
[prompt] * batch_size,
417417
num_inference_steps=2,
418418
output_type="np",
419-
init_image=init_image,
419+
image=init_image,
420420
num_images_per_prompt=num_images_per_prompt,
421421
).images
422422

@@ -458,7 +458,7 @@ def test_stable_diffusion_img2img_fp16(self):
458458
generator=generator,
459459
num_inference_steps=2,
460460
output_type="np",
461-
init_image=init_image,
461+
image=init_image,
462462
).images
463463

464464
assert image.shape == (1, 32, 32, 3)
@@ -497,7 +497,7 @@ def test_stable_diffusion_img2img_pipeline_default(self):
497497
generator = torch.Generator(device=torch_device).manual_seed(0)
498498
output = pipe(
499499
prompt=prompt,
500-
init_image=init_image,
500+
image=init_image,
501501
strength=0.75,
502502
guidance_scale=7.5,
503503
generator=generator,
@@ -535,7 +535,7 @@ def test_stable_diffusion_img2img_pipeline_k_lms(self):
535535
generator = torch.Generator(device=torch_device).manual_seed(0)
536536
output = pipe(
537537
prompt=prompt,
538-
init_image=init_image,
538+
image=init_image,
539539
strength=0.75,
540540
guidance_scale=7.5,
541541
generator=generator,
@@ -572,7 +572,7 @@ def test_stable_diffusion_img2img_pipeline_ddim(self):
572572
generator = torch.Generator(device=torch_device).manual_seed(0)
573573
output = pipe(
574574
prompt=prompt,
575-
init_image=init_image,
575+
image=init_image,
576576
strength=0.75,
577577
guidance_scale=7.5,
578578
generator=generator,
@@ -626,7 +626,7 @@ def test_callback_fn(step: int, timestep: int, latents: torch.FloatTensor) -> No
626626
with torch.autocast(torch_device):
627627
pipe(
628628
prompt=prompt,
629-
init_image=init_image,
629+
image=init_image,
630630
strength=0.75,
631631
num_inference_steps=50,
632632
guidance_scale=7.5,
@@ -663,7 +663,7 @@ def test_stable_diffusion_pipeline_with_sequential_cpu_offloading(self):
663663
generator = torch.Generator(device=torch_device).manual_seed(0)
664664
_ = pipe(
665665
prompt=prompt,
666-
init_image=init_image,
666+
image=init_image,
667667
strength=0.75,
668668
guidance_scale=7.5,
669669
generator=generator,

‎tests/pipelines/stable_diffusion/test_stable_diffusion_inpaint_legacy.py‎

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -191,7 +191,7 @@ def test_stable_diffusion_inpaint_legacy(self):
191191
guidance_scale=6.0,
192192
num_inference_steps=2,
193193
output_type="np",
194-
init_image=init_image,
194+
image=init_image,
195195
mask_image=mask_image,
196196
)
197197

@@ -204,7 +204,7 @@ def test_stable_diffusion_inpaint_legacy(self):
204204
guidance_scale=6.0,
205205
num_inference_steps=2,
206206
output_type="np",
207-
init_image=init_image,
207+
image=init_image,
208208
mask_image=mask_image,
209209
return_dict=False,
210210
)[0]
@@ -252,7 +252,7 @@ def test_stable_diffusion_inpaint_legacy_negative_prompt(self):
252252
guidance_scale=6.0,
253253
num_inference_steps=2,
254254
output_type="np",
255-
init_image=init_image,
255+
image=init_image,
256256
mask_image=mask_image,
257257
)
258258

@@ -295,7 +295,7 @@ def test_stable_diffusion_inpaint_legacy_num_images_per_prompt(self):
295295
prompt,
296296
num_inference_steps=2,
297297
output_type="np",
298-
init_image=init_image,
298+
image=init_image,
299299
mask_image=mask_image,
300300
).images
301301

@@ -307,7 +307,7 @@ def test_stable_diffusion_inpaint_legacy_num_images_per_prompt(self):
307307
[prompt] * batch_size,
308308
num_inference_steps=2,
309309
output_type="np",
310-
init_image=init_image,
310+
image=init_image,
311311
mask_image=mask_image,
312312
).images
313313

@@ -319,7 +319,7 @@ def test_stable_diffusion_inpaint_legacy_num_images_per_prompt(self):
319319
prompt,
320320
num_inference_steps=2,
321321
output_type="np",
322-
init_image=init_image,
322+
image=init_image,
323323
mask_image=mask_image,
324324
num_images_per_prompt=num_images_per_prompt,
325325
).images
@@ -332,7 +332,7 @@ def test_stable_diffusion_inpaint_legacy_num_images_per_prompt(self):
332332
[prompt] * batch_size,
333333
num_inference_steps=2,
334334
output_type="np",
335-
init_image=init_image,
335+
image=init_image,
336336
mask_image=mask_image,
337337
num_images_per_prompt=num_images_per_prompt,
338338
).images
@@ -374,7 +374,7 @@ def test_stable_diffusion_inpaint_legacy_pipeline(self):
374374
generator = torch.Generator(device=torch_device).manual_seed(0)
375375
output = pipe(
376376
prompt=prompt,
377-
init_image=init_image,
377+
image=init_image,
378378
mask_image=mask_image,
379379
strength=0.75,
380380
guidance_scale=7.5,
@@ -416,7 +416,7 @@ def test_stable_diffusion_inpaint_legacy_pipeline_k_lms(self):
416416
generator = torch.Generator(device=torch_device).manual_seed(0)
417417
output = pipe(
418418
prompt=prompt,
419-
init_image=init_image,
419+
image=init_image,
420420
mask_image=mask_image,
421421
strength=0.75,
422422
guidance_scale=7.5,
@@ -474,7 +474,7 @@ def test_callback_fn(step: int, timestep: int, latents: torch.FloatTensor) -> No
474474
with torch.autocast(torch_device):
475475
pipe(
476476
prompt=prompt,
477-
init_image=init_image,
477+
image=init_image,
478478
mask_image=mask_image,
479479
strength=0.75,
480480
num_inference_steps=50,

‎tests/test_pipelines.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -395,15 +395,15 @@ def test_components(self):
395395
generator=generator,
396396
num_inference_steps=2,
397397
output_type="np",
398-
init_image=init_image,
398+
image=init_image,
399399
mask_image=mask_image,
400400
).images
401401
image_img2img = img2img(
402402
[prompt],
403403
generator=generator,
404404
num_inference_steps=2,
405405
output_type="np",
406-
init_image=init_image,
406+
image=init_image,
407407
).images
408408
image_text2img = text2img(
409409
[prompt],

0 commit comments

Comments
 (0)