Skip to content

Commit d647900

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

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
@@ -537,8 +537,8 @@ def test_stable_diffusion_pipeline_with_sequential_cpu_offloading(self):
537537

538538
def test_stable_diffusion_pipeline_with_model_offloading(self):
539539
backend_empty_cache(torch_device)
540-
backend_reset_peak_memory_stats(torch_device)
541540
backend_reset_max_memory_allocated(torch_device)
541+
backend_reset_peak_memory_stats(torch_device)
542542

543543
inputs = self.get_inputs(torch_device, dtype=torch.float16)
544544

0 commit comments

Comments
 (0)