end_latent

app.py

@@ -696,9 +696,9 @@ def worker_start_end(input_image, end_image, image_position, prompts, n_prompt,
                     image_encoder
                 )
                 
-            return [start_latent, image_encoder_last_hidden_state]
+            return [start_latent, end_latent, image_encoder_last_hidden_state]
             
-        [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)
+        [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)
 
         # Dtype
         image_encoder_last_hidden_state = image_encoder_last_hidden_state.to(transformer.dtype)