Earlier
Browse files
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))
|