Update app.py

app.py

@@ -173,6 +173,15 @@ def tiled_flux_img2img(pipe, prompt, image, strength, steps, guidance, generator
     w, h = image.size
     output = image.copy()  # Start with the control image
 
+    # For handling long prompts: truncate for CLIP, full for T5
+    max_clip_tokens = pipe.tokenizer.model_max_length  # Typically 77
+    input_ids = pipe.tokenizer.encode(prompt, return_tensors="pt")
+    if input_ids.shape[1] > max_clip_tokens:
+        input_ids = input_ids[:, :max_clip_tokens]
+        prompt_clip = pipe.tokenizer.decode(input_ids[0], skip_special_tokens=True)
+    else:
+        prompt_clip = prompt
+
     for x in range(0, w, tile_size - overlap):
         for y in range(0, h, tile_size - overlap):
             tile_w = min(tile_size, w - x)
@@ -181,7 +190,8 @@ def tiled_flux_img2img(pipe, prompt, image, strength, steps, guidance, generator
 
             # Run Flux on tile
             gen_tile = pipe(
-                prompt=prompt,
+                prompt=prompt_clip,
+                prompt_2=prompt,
                 image=tile,
                 strength=strength,
                 num_inference_steps=steps,