Update tests/pipelines/stable_diffusion/test_stable_diffusion.py

Co-authored-by: hlky <[email protected]>

Author: faaany

tests/pipelines/stable_diffusion/test_stable_diffusion.py

@@ -1117,6 +1117,7 @@ def test_stable_diffusion_pipeline_with_model_offloading(self):
         pipe.unet.set_default_attn_processor()
 
         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)