Update ui.py

ui.py

@@ -74,6 +74,8 @@ def generate_multiple_images_batch(
             height=height,
             num_inference_steps=num_inference_steps,
             generator=generator,
+            max_sequence_length=256,  # Maximum allowed for schnell
+            truncation=True,  # Add this parameter
             guidance_scale=0.0
         ).images