1inkusFace commited on
Commit
f663c13
·
verified ·
1 Parent(s): 05c2c71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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[0])
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[0])
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[0])
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]