Update app.py

app.py

@@ -128,7 +128,6 @@ def load_and_prepare_model():
         #torch_dtype=torch.bfloat16,
         add_watermarker=False,
         token=True,
-        add_watermarker=False,
         text_encoder=None,
         text_encoder_2=None,
         tokenizer=None,
@@ -251,6 +250,9 @@ def generate_30(
     #pipe.set_adapters(["skin"], adapter_weights=[lora_scale])
     seed = int(randomize_seed_fn())
     generator = torch.Generator(device='cuda').manual_seed(seed)
+
+    pipe.text_encoder=text_encoder.to(device=device, dtype=torch.bfloat16)
+    pipe.text_encoder_2=text_encoder_2.to(device=device, dtype=torch.bfloat16)
     #prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
     options = {
         "prompt": [prompt],
@@ -306,6 +308,9 @@ def generate_60(
     #pipe.set_adapters(["skin"], adapter_weights=[lora_scale])
     seed = int(randomize_seed_fn())
     generator = torch.Generator(device='cuda').manual_seed(seed)
+
+    pipe.text_encoder=text_encoder.to(device=device, dtype=torch.bfloat16)
+    pipe.text_encoder_2=text_encoder_2.to(device=device, dtype=torch.bfloat16)
     options = {
         "prompt": [prompt],
         "negative_prompt": [negative_prompt],
@@ -358,6 +363,9 @@ def generate_90(
     #pipe.set_adapters(["skin"], adapter_weights=[lora_scale])
     seed = int(randomize_seed_fn())
     generator = torch.Generator(device='cuda').manual_seed(seed)
+
+    pipe.text_encoder=text_encoder.to(device=device, dtype=torch.bfloat16)
+    pipe.text_encoder_2=text_encoder_2.to(device=device, dtype=torch.bfloat16)
     options = {
         "prompt": [prompt],
         "negative_prompt": [negative_prompt],