Update tests/pipelines/stable_diffusion/test_stable_diffusion_img2img.py

faaany · hlky · web-flow · commit cfbf6019a2c5 · 2025-02-27T15:06:34.000+08:00
Co-authored-by: hlky &lt;hlky@hlky.ac&gt;
diff --git a/tests/pipelines/stable_diffusion/test_stable_diffusion_img2img.py b/tests/pipelines/stable_diffusion/test_stable_diffusion_img2img.py
@@ -564,6 +564,7 @@ def test_stable_diffusion_pipeline_with_model_offloading(self):
         )
 
         backend_empty_cache(torch_device)
+        backend_reset_max_memory_allocated(torch_device)
         backend_reset_peak_memory_stats(torch_device)
 
         pipe.enable_model_cpu_offload(device=torch_device)

Original file line number	Diff line number	Diff line change
`@@ -564,6 +564,7 @@ def test_stable_diffusion_pipeline_with_model_offloading(self):`
`564`	`564`	`)`
`565`	`565`
`566`	`566`	`backend_empty_cache(torch_device)`
	`567`	`+ backend_reset_max_memory_allocated(torch_device)`
`567`	`568`	`backend_reset_peak_memory_stats(torch_device)`
`568`	`569`
`569`	`570`	`pipe.enable_model_cpu_offload(device=torch_device)`