We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 55f9658 commit d647900Copy full SHA for d647900
tests/pipelines/stable_diffusion/test_stable_diffusion_img2img.py
@@ -537,8 +537,8 @@ def test_stable_diffusion_pipeline_with_sequential_cpu_offloading(self):
537
538
def test_stable_diffusion_pipeline_with_model_offloading(self):
539
backend_empty_cache(torch_device)
540
- backend_reset_peak_memory_stats(torch_device)
541
backend_reset_max_memory_allocated(torch_device)
+ backend_reset_peak_memory_stats(torch_device)
542
543
inputs = self.get_inputs(torch_device, dtype=torch.float16)
544
0 commit comments