RealVis_v5.0_BF16_F

Runtime error

1inkusFace commited on Jan 21

Commit

4e598c3

verified ·

1 Parent(s): 75cd14f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -209,8 +209,8 @@ def generate_30(
                     return_tensors="pt",
     )
     text_input_ids2 = text_inputs2.input_ids
-    prompt_embedsa = pipe.text_encoder(text_input_ids1.to(device, dtype=torch.bfloat16), output_hidden_states=True)
-    prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device, dtype=torch.bfloat16), output_hidden_states=True)
     prompt_embeds = torch.cat([prompt_embedsa,prompt_embedsb]).mean(dim=-1)
     options = {
@@ -274,8 +274,8 @@ def generate_60(
                     return_tensors="pt",
     )
     text_input_ids2 = text_inputs2.input_ids
-    prompt_embedsa = pipe.text_encoder(text_input_ids1.to(device, dtype=torch.bfloat16), output_hidden_states=True)
-    prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device, dtype=torch.bfloat16), output_hidden_states=True)
     prompt_embeds = torch.cat([prompt_embedsa,prompt_embedsb]).mean(dim=-1)
     options = {
@@ -339,8 +339,8 @@ def generate_90(
                     return_tensors="pt",
     )
     text_input_ids2 = text_inputs2.input_ids
-    prompt_embedsa = pipe.text_encoder(text_input_ids1.to(device, dtype=torch.bfloat16), output_hidden_states=True)
-    prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device, dtype=torch.bfloat16), output_hidden_states=True)
     prompt_embeds = torch.cat([prompt_embedsa,prompt_embedsb]).mean(dim=-1)
     options = {

                     return_tensors="pt",
     )
     text_input_ids2 = text_inputs2.input_ids
+    prompt_embedsa = pipe.text_encoder(text_input_ids1.to(device), output_hidden_states=True)
+    prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
     prompt_embeds = torch.cat([prompt_embedsa,prompt_embedsb]).mean(dim=-1)
     options = {
                     return_tensors="pt",
     )
     text_input_ids2 = text_inputs2.input_ids
+    prompt_embedsa = pipe.text_encoder(text_input_ids1.to(device), output_hidden_states=True)
+    prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
     prompt_embeds = torch.cat([prompt_embedsa,prompt_embedsb]).mean(dim=-1)
     options = {
                     return_tensors="pt",
     )
     text_input_ids2 = text_inputs2.input_ids
+    prompt_embedsa = pipe.text_encoder(text_input_ids1.to(device), output_hidden_states=True)
+    prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
     prompt_embeds = torch.cat([prompt_embedsa,prompt_embedsb]).mean(dim=-1)
     options = {