y

app.py

@@ -158,6 +158,8 @@ def infer(
                 variant="fp16",
                 use_safetensors=True
             ).to(device)
+            pipe.text_encoder = pipe.text_encoder.to(device)
+            pipe.text_encoder_2 = pipe.text_encoder_2.to(device)
 
         # Reproducibility
         if seed != -1: