Update app.py

app.py

@@ -26,7 +26,7 @@ if torch.cuda.is_available():
         add_watermarker=False,
     )
     # pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
-    pipe.tokenizer.model_max_length = 512
+    # pipe.tokenizer.model_max_length = 512
     pipe.to("cuda")
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
@@ -77,10 +77,10 @@ def infer(
         negative_pooled_prompt_embeds = None
 
     image = pipe(
-        prompt_embeds=prompt_embeds,
-        negative_prompt_embeds=prompt_neg_embeds,
-        pooled_prompt_embeds=pooled_prompt_embeds,
-        negative_pooled_prompt_embeds=negative_pooled_prompt_embeds,
+        prompt=prompt_embeds,
+        negative_prompt=prompt_neg_embeds,
+        #pooled_prompt_embeds=pooled_prompt_embeds,
+        #negative_pooled_prompt_embeds=negative_pooled_prompt_embeds,
         width=width,
         height=height,
         guidance_scale=guidance_scale,