Update app.py

app.py

@@ -89,11 +89,13 @@ def generate(segment, image, prompt, size, guidance_scale, num_inference_steps,
         #    config=offload_config,
         #)
         pipe.text_encoder.to("cuda")
+        pipe.text_encoder_2.to("cuda")
         with torch.no_grad():
             prompt_embeds, prompt_attention_mask, negative_prompt_embeds, negative_attention_mask, pooled_prompt_embeds, negative_pooled_prompt_embeds = pipe.encode_prompt(
                 prompt=prompt, do_classifier_free_guidance=True, negative_prompt=negative_prompt, device=device
             )
         pipe.text_encoder.to("cpu")
+        pipe.text_encoder_2.to("cpu")
         transformer_dtype = pipe.transformer.dtype
         prompt_embeds = prompt_embeds.to(transformer_dtype)
         prompt_attention_mask = prompt_attention_mask.to(transformer_dtype)