Update app.py

app.py

@@ -70,6 +70,8 @@ def encode_prompt(prompt_batch, text_encoder, tokenizer, proportion_empty_prompt
             truncation=True,
             return_tensors="pt",
         )
+        for name, param in text_encoder.named_parameters():
+            print(name, param.device)
 
         print(f"Text Encoder Device: {text_encoder.device}")
         text_input_ids = text_inputs.input_ids.cuda()