stable unclip integration tests turn on memory saving (huggingface#2408)

* stable unclip integration tests turn on memory saving

* add note on turning on memory saving

Author: williamberman

tests/pipelines/stable_unclip/test_stable_unclip.py

@@ -189,6 +189,10 @@ def test_stable_unclip(self):
         pipe = StableUnCLIPPipeline.from_pretrained("fusing/stable-unclip-2-1-l", torch_dtype=torch.float16)
         pipe.to(torch_device)
         pipe.set_progress_bar_config(disable=None)
+        # stable unclip will oom when integration tests are run on a V100,
+        # so turn on memory savings
+        pipe.enable_attention_slicing()
+        pipe.enable_sequential_cpu_offload()
 
         generator = torch.Generator(device="cpu").manual_seed(0)
         output = pipe("anime turle", generator=generator, output_type="np")

tests/pipelines/stable_unclip/test_stable_unclip_img2img.py

@@ -185,6 +185,10 @@ def test_stable_unclip_l_img2img(self):
         )
         pipe.to(torch_device)
         pipe.set_progress_bar_config(disable=None)
+        # stable unclip will oom when integration tests are run on a V100,
+        # so turn on memory savings
+        pipe.enable_attention_slicing()
+        pipe.enable_sequential_cpu_offload()
 
         generator = torch.Generator(device="cpu").manual_seed(0)
         output = pipe("anime turle", image=input_image, generator=generator, output_type="np")
@@ -209,6 +213,10 @@ def test_stable_unclip_h_img2img(self):
         )
         pipe.to(torch_device)
         pipe.set_progress_bar_config(disable=None)
+        # stable unclip will oom when integration tests are run on a V100,
+        # so turn on memory savings
+        pipe.enable_attention_slicing()
+        pipe.enable_sequential_cpu_offload()
 
         generator = torch.Generator(device="cpu").manual_seed(0)
         output = pipe("anime turle", image=input_image, generator=generator, output_type="np")