Update app.py

app.py

@@ -185,8 +185,8 @@ def generate_30(
 ):
     seed = random.randint(0, MAX_SEED)
     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)
+    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],
@@ -227,8 +227,8 @@ def generate_60(
 ):
     seed = random.randint(0, MAX_SEED)
     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)
+    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],
@@ -269,8 +269,8 @@ def generate_90(
 ):
     seed = random.randint(0, MAX_SEED)
     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)
+    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],