Liu Hanchen
commited on
Commit
·
4a3d4ad
1
Parent(s):
d11262e
Update Sci_Fi_inbetweening_pipeline.py
Browse files
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=
|
| 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)
|