Skip to content

Commit 88263e8

Browse files
faaanyhlky
andauthored
Update tests/pipelines/stable_diffusion/test_stable_diffusion_img2img.py
Co-authored-by: hlky <[email protected]>
1 parent cfbf601 commit 88263e8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/pipelines/stable_diffusion/test_stable_diffusion_img2img.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -518,8 +518,8 @@ def callback_fn(step: int, timestep: int, latents: torch.Tensor) -> None:
518518

519519
def test_stable_diffusion_pipeline_with_sequential_cpu_offloading(self):
520520
backend_empty_cache(torch_device)
521-
backend_reset_peak_memory_stats(torch_device)
522521
backend_reset_max_memory_allocated(torch_device)
522+
backend_reset_peak_memory_stats(torch_device)
523523

524524
pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
525525
"CompVis/stable-diffusion-v1-4", safety_checker=None, torch_dtype=torch.float16

0 commit comments

Comments
 (0)