Update app.py

app.py

@@ -148,7 +148,7 @@ def infer(
     outputs = model.generate(
         input_ids=input_ids,
         attention_mask=attention_mask,
-        max_new_tokens=1024,
+        max_new_tokens=512,
         temperature=0.2,
         top_p=0.9,
         do_sample=True,
@@ -211,7 +211,7 @@ def infer(
             height=height,
          #   latents=None,
             generator=generator,
-            max_sequence_length=1024
+            max_sequence_length=512
         ).images[0]
         print('-- got image --')
     image_path = f"sd35m_{seed}.png"