Update pipeline.py

pipeline.py

@@ -119,8 +119,14 @@ class StableDiffusionComparisonPipeline(DiffusionPipeline):
         if slice_size == "auto":
             # half the attention head size is usually a good trade-off between
             # speed and memory
-            slice_size = self.unet.config.attention_head_dim // 2
-        self.unet.set_attention_slice(slice_size)
+            slice_size_1 = self.pipe1.unet.config.attention_head_dim // 2
+            slice_size_2 = self.pipe2.unet.config.attention_head_dim // 2
+            slice_size_3 = self.pipe3.unet.config.attention_head_dim // 2
+            slice_size_4 = self.pipe4.unet.config.attention_head_dim // 2
+        self.pipe1.unet.set_attention_slice(slice_size_1)
+        self.pipe2.unet.set_attention_slice(slice_size_2)
+        self.pipe3.unet.set_attention_slice(slice_size_3)
+        self.pipe4.unet.set_attention_slice(slice_size_4)
 
     def disable_attention_slicing(self):
         r"""