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

@@ -1139,6 +1139,7 @@ def test_stable_diffusion_pipeline_with_model_offloading(self):
 
         # With attention slicing
         backend_empty_cache(torch_device)
+        backend_reset_max_memory_allocated(torch_device)
         backend_reset_peak_memory_stats(torch_device)
 
         pipe.enable_attention_slicing()