Liu Hanchen commited on
Commit
4a3d4ad
·
1 Parent(s): d11262e

Update Sci_Fi_inbetweening_pipeline.py

Browse files
Files changed (1) hide show
  1. Sci_Fi_inbetweening_pipeline.py +2 -2
Sci_Fi_inbetweening_pipeline.py CHANGED
@@ -743,7 +743,7 @@ class CogVideoXEFNetInbetweeningPipeline(DiffusionPipeline, CogVideoXLoraLoaderM
743
  if (EF_Net_guidance_start <= current_sampling_percent < EF_Net_guidance_end):
744
  # extract EF_Net hidden state
745
  EF_Net_states = self.EF_Net_model(
746
- hidden_states=latent_image_input[:,:,0:16,:,:],
747
  encoder_hidden_states=prompt_embeds,
748
  image_rotary_emb=None,
749
  EF_Net_states=latent_image_input[:,12::,:,:,:],
@@ -821,4 +821,4 @@ class CogVideoXEFNetInbetweeningPipeline(DiffusionPipeline, CogVideoXLoraLoaderM
821
  if not return_dict:
822
  return (video,)
823
 
824
- return CogVideoXPipelineOutput(frames=video)
 
743
  if (EF_Net_guidance_start <= current_sampling_percent < EF_Net_guidance_end):
744
  # extract EF_Net hidden state
745
  EF_Net_states = self.EF_Net_model(
746
+ hidden_states=latent_model_input[:,:,0:16,:,:],
747
  encoder_hidden_states=prompt_embeds,
748
  image_rotary_emb=None,
749
  EF_Net_states=latent_image_input[:,12::,:,:,:],
 
821
  if not return_dict:
822
  return (video,)
823
 
824
+ return CogVideoXPipelineOutput(frames=video)