Update tests/pipelines/controlnet/test_controlnet.py

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

Author: faaany

tests/pipelines/controlnet/test_controlnet.py

@@ -951,10 +951,7 @@ def test_sequential_cpu_offloading(self):
             output_type="np",
         )
 
-        if torch_device == "cuda":
-            mem_bytes = torch.cuda.max_memory_allocated()
-        elif torch_device == "xpu":
-            mem_bytes = torch.xpu.max_memory_allocated()
+        mem_bytes = backend_max_memory_allocated(torch_device)
         # make sure that less than 7 GB is allocated
         assert mem_bytes < 4 * 10**9