Update pipelines/pipeline_seesr.py

pipelines/pipeline_seesr.py

@@ -1097,9 +1097,9 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
                         image_encoder_hidden_states = ram_encoder_hidden_states,
                     )[0]
                 else:
-                    tile_weights = self._gaussian_weights(tile_size, tile_size, 1)
+                    tile_weights = self._gaussian_weights(tile_size, tile_size, batch_size)
                     tile_size = min(tile_size, min(h, w))
-                    tile_weights = self._gaussian_weights(tile_size, tile_size, 1)
+                    tile_weights = self._gaussian_weights(tile_size, tile_size, batch_size)
 
                     grid_rows = 0
                     cur_x = 0