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=240,
+        max_new_tokens=1024,
         temperature=0.2,
         top_p=0.9,
         do_sample=True,
@@ -201,9 +201,9 @@ def infer(
         print('-- generating image --')
         #with torch.no_grad():
         sd_image = pipe(
-            prompt=enhanced_prompt,  # This conversion is fine
+            prompt=prompt,  # This conversion is fine
             prompt_2=enhanced_prompt_2,
-            prompt_3=prompt,
+            prompt_3=enhanced_prompt,
             negative_prompt=negative_prompt,
             guidance_scale=guidance_scale,
             num_inference_steps=num_inference_steps,
@@ -211,6 +211,7 @@ def infer(
             height=height,
          #   latents=None,
             generator=generator,
+            max_sequence_length=1024
         ).images[0]
         print('-- got image --')
     image_path = f"sd35m_{seed}.png"