1inkusFace commited on
Commit
dcc9828
·
verified ·
1 Parent(s): 6c95eb1

Update app.py

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