Fabrice-TIERCELIN commited on
Commit
bc1599f
·
verified ·
1 Parent(s): 2cb41fe
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -175,6 +175,7 @@ def worker(input_image, prompt, n_prompt, seed, total_second_length, latent_wind
175
  image_encoder_last_hidden_state = image_encoder_output.last_hidden_state
176
 
177
  # Dtype
 
178
 
179
  image_encoder_last_hidden_state = image_encoder_last_hidden_state.to(transformer.dtype)
180
 
@@ -190,9 +191,6 @@ def worker(input_image, prompt, n_prompt, seed, total_second_length, latent_wind
190
  history_latents = torch.cat([history_latents, start_latent.to(history_latents)], dim=2)
191
  total_generated_latent_frames = 1
192
 
193
- if len(prompt_parameters) > 1:
194
- [llama_vec, clip_l_pooler, llama_vec_n, clip_l_pooler_n, llama_attention_mask, llama_attention_mask_n] = prompt_parameters.pop(0)
195
-
196
  for section_index in range(total_latent_sections):
197
  if stream.input_queue.top() == 'end':
198
  stream.output_queue.push(('end', None))
 
175
  image_encoder_last_hidden_state = image_encoder_output.last_hidden_state
176
 
177
  # Dtype
178
+ [llama_vec, clip_l_pooler, llama_vec_n, clip_l_pooler_n, llama_attention_mask, llama_attention_mask_n] = prompt_parameters.pop(0)
179
 
180
  image_encoder_last_hidden_state = image_encoder_last_hidden_state.to(transformer.dtype)
181
 
 
191
  history_latents = torch.cat([history_latents, start_latent.to(history_latents)], dim=2)
192
  total_generated_latent_frames = 1
193
 
 
 
 
194
  for section_index in range(total_latent_sections):
195
  if stream.input_queue.top() == 'end':
196
  stream.output_queue.push(('end', None))