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 |
-
print('text_encoder shape: ',prompt_embedsa.shape)
|
216 |
pooled_prompt_embeds_list.append(prompt_embedsa[0])
|
217 |
prompt_embedsa = prompt_embedsa.hidden_states[-2]
|
|
|
218 |
prompt_embeds_list.append(prompt_embedsa)
|
219 |
prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
|
220 |
pooled_prompt_embeds_list.append(prompt_embedsb[0])
|
@@ -289,9 +289,9 @@ def generate_60(
|
|
289 |
)
|
290 |
text_input_ids2 = text_inputs2.input_ids
|
291 |
prompt_embedsa = pipe.text_encoder(text_input_ids1.to(device), output_hidden_states=True)
|
292 |
-
print('text_encoder shape: ',prompt_embedsa.shape)
|
293 |
pooled_prompt_embeds_list.append(prompt_embedsa[0])
|
294 |
prompt_embedsa = prompt_embedsa.hidden_states[-2]
|
|
|
295 |
prompt_embeds_list.append(prompt_embedsa)
|
296 |
prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
|
297 |
pooled_prompt_embeds_list.append(prompt_embedsb[0])
|
@@ -366,9 +366,9 @@ def generate_90(
|
|
366 |
)
|
367 |
text_input_ids2 = text_inputs2.input_ids
|
368 |
prompt_embedsa = pipe.text_encoder(text_input_ids1.to(device), output_hidden_states=True)
|
369 |
-
print('text_encoder shape: ',prompt_embedsa.shape)
|
370 |
pooled_prompt_embeds_list.append(prompt_embedsa[0])
|
371 |
prompt_embedsa = prompt_embedsa.hidden_states[-2]
|
|
|
372 |
prompt_embeds_list.append(prompt_embedsa)
|
373 |
prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
|
374 |
pooled_prompt_embeds_list.append(prompt_embedsb[0])
|
|
|
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 |
+
print('text_encoder shape: ',prompt_embedsa.shape)
|
218 |
prompt_embeds_list.append(prompt_embedsa)
|
219 |
prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
|
220 |
pooled_prompt_embeds_list.append(prompt_embedsb[0])
|
|
|
289 |
)
|
290 |
text_input_ids2 = text_inputs2.input_ids
|
291 |
prompt_embedsa = pipe.text_encoder(text_input_ids1.to(device), output_hidden_states=True)
|
|
|
292 |
pooled_prompt_embeds_list.append(prompt_embedsa[0])
|
293 |
prompt_embedsa = prompt_embedsa.hidden_states[-2]
|
294 |
+
print('text_encoder shape: ',prompt_embedsa.shape)
|
295 |
prompt_embeds_list.append(prompt_embedsa)
|
296 |
prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
|
297 |
pooled_prompt_embeds_list.append(prompt_embedsb[0])
|
|
|
366 |
)
|
367 |
text_input_ids2 = text_inputs2.input_ids
|
368 |
prompt_embedsa = pipe.text_encoder(text_input_ids1.to(device), output_hidden_states=True)
|
|
|
369 |
pooled_prompt_embeds_list.append(prompt_embedsa[0])
|
370 |
prompt_embedsa = prompt_embedsa.hidden_states[-2]
|
371 |
+
print('text_encoder shape: ',prompt_embedsa.shape)
|
372 |
prompt_embeds_list.append(prompt_embedsa)
|
373 |
prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
|
374 |
pooled_prompt_embeds_list.append(prompt_embedsb[0])
|