Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -212,9 +212,9 @@ def generate_30(
|
|
212 |
)
|
213 |
text_input_ids2 = text_inputs2.input_ids
|
214 |
prompt_embedsa = pipe.text_encoder(text_input_ids1.to(device), output_hidden_states=True)
|
215 |
-
pooled_prompt_embeds_list.append(prompt_embedsa)
|
216 |
prompt_embedsa = prompt_embedsa.hidden_states[-2]
|
217 |
-
prompt_embeds_list.append(prompt_embedsa
|
218 |
prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
|
219 |
pooled_prompt_embeds_list.append(prompt_embedsb[0])
|
220 |
prompt_embedsb = prompt_embedsb.hidden_states[-2]
|
@@ -288,9 +288,9 @@ def generate_60(
|
|
288 |
)
|
289 |
text_input_ids2 = text_inputs2.input_ids
|
290 |
prompt_embedsa = pipe.text_encoder(text_input_ids1.to(device), output_hidden_states=True)
|
291 |
-
pooled_prompt_embeds_list.append(prompt_embedsa)
|
292 |
prompt_embedsa = prompt_embedsa.hidden_states[-2]
|
293 |
-
prompt_embeds_list.append(prompt_embedsa
|
294 |
prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
|
295 |
pooled_prompt_embeds_list.append(prompt_embedsb[0])
|
296 |
prompt_embedsb = prompt_embedsb.hidden_states[-2]
|
@@ -364,9 +364,9 @@ def generate_90(
|
|
364 |
)
|
365 |
text_input_ids2 = text_inputs2.input_ids
|
366 |
prompt_embedsa = pipe.text_encoder(text_input_ids1.to(device), output_hidden_states=True)
|
367 |
-
pooled_prompt_embeds_list.append(prompt_embedsa)
|
368 |
prompt_embedsa = prompt_embedsa.hidden_states[-2]
|
369 |
-
prompt_embeds_list.append(prompt_embedsa
|
370 |
prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
|
371 |
pooled_prompt_embeds_list.append(prompt_embedsb[0])
|
372 |
prompt_embedsb = prompt_embedsb.hidden_states[-2]
|
|
|
212 |
)
|
213 |
text_input_ids2 = text_inputs2.input_ids
|
214 |
prompt_embedsa = pipe.text_encoder(text_input_ids1.to(device), output_hidden_states=True)
|
215 |
+
pooled_prompt_embeds_list.append(prompt_embedsa[0])
|
216 |
prompt_embedsa = prompt_embedsa.hidden_states[-2]
|
217 |
+
prompt_embeds_list.append(prompt_embedsa)
|
218 |
prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
|
219 |
pooled_prompt_embeds_list.append(prompt_embedsb[0])
|
220 |
prompt_embedsb = prompt_embedsb.hidden_states[-2]
|
|
|
288 |
)
|
289 |
text_input_ids2 = text_inputs2.input_ids
|
290 |
prompt_embedsa = pipe.text_encoder(text_input_ids1.to(device), output_hidden_states=True)
|
291 |
+
pooled_prompt_embeds_list.append(prompt_embedsa[0])
|
292 |
prompt_embedsa = prompt_embedsa.hidden_states[-2]
|
293 |
+
prompt_embeds_list.append(prompt_embedsa)
|
294 |
prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
|
295 |
pooled_prompt_embeds_list.append(prompt_embedsb[0])
|
296 |
prompt_embedsb = prompt_embedsb.hidden_states[-2]
|
|
|
364 |
)
|
365 |
text_input_ids2 = text_inputs2.input_ids
|
366 |
prompt_embedsa = pipe.text_encoder(text_input_ids1.to(device), output_hidden_states=True)
|
367 |
+
pooled_prompt_embeds_list.append(prompt_embedsa[0])
|
368 |
prompt_embedsa = prompt_embedsa.hidden_states[-2]
|
369 |
+
prompt_embeds_list.append(prompt_embedsa)
|
370 |
prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
|
371 |
pooled_prompt_embeds_list.append(prompt_embedsb[0])
|
372 |
prompt_embedsb = prompt_embedsb.hidden_states[-2]
|