Update generator.py

generator.py

@@ -772,7 +772,7 @@ class RetroArtConverter:
                 conditioning_batch, pooled_batch = self.compel([prompt, negative_prompt])
 
                 # Unpack the batch results using slicing
-                # [0:1] keeps the batch dimension, which is required
+                # [0:1] and [1:2] keeps the batch dimension, which is required
                 pipe_kwargs["prompt_embeds"] = conditioning_batch[0:1]
                 pipe_kwargs["pooled_prompt_embeds"] = pooled_batch[0:1]
                 pipe_kwargs["negative_prompt_embeds"] = conditioning_batch[1:2]