Fabrice-TIERCELIN commited on
Commit
e652791
·
verified ·
1 Parent(s): 9fd053d

end_latent

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