Spaces:
Running
Running
end_latent
Browse files
app.py
CHANGED
@@ -696,9 +696,9 @@ def worker_start_end(input_image, end_image, image_position, prompts, n_prompt,
|
|
696 |
image_encoder
|
697 |
)
|
698 |
|
699 |
-
return [start_latent, image_encoder_last_hidden_state]
|
700 |
|
701 |
-
[start_latent, image_encoder_last_hidden_state] = get_start_latent(input_image, has_end_image, end_image, height, width, vae, gpu, image_encoder, high_vram)
|
702 |
|
703 |
# Dtype
|
704 |
image_encoder_last_hidden_state = image_encoder_last_hidden_state.to(transformer.dtype)
|
|
|
696 |
image_encoder
|
697 |
)
|
698 |
|
699 |
+
return [start_latent, end_latent, image_encoder_last_hidden_state]
|
700 |
|
701 |
+
[start_latent, end_latent, image_encoder_last_hidden_state] = get_start_latent(input_image, has_end_image, end_image, height, width, vae, gpu, image_encoder, high_vram)
|
702 |
|
703 |
# Dtype
|
704 |
image_encoder_last_hidden_state = image_encoder_last_hidden_state.to(transformer.dtype)
|