Update app.py

app.py

@@ -54,6 +54,10 @@ def infer(
     prompt = str(prompt) if prompt else ""
     negative_prompt = str(negative_prompt) if negative_prompt else ""
 
+    # Ensure text input IDs are of type LongTensor
+    if hasattr(pipe.text_encoder, 'input_ids'):
+        pipe.text_encoder.input_ids = pipe.text_encoder.input_ids.to(torch.long)
+
     image = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
@@ -166,4 +170,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()