theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-24 10:18:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #984 from modelscope/animate-bugfix

Browse Source 
bugfix
This commit is contained in:
Zhongjie Duan
2025-10-10 15:42:20 +08:00
committed by GitHub
parent 084bc2fc78 77fac2a03f
commit 0a1c172a00
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
diffsynth/pipelines/wan_video_new.py
View File

@@ -1102,7 +1102,7 @@ class WanVideoPostUnit_AnimateFacePixelValues(PipelineUnit):
def process(self, pipe: WanVideoPipeline, inputs_shared, inputs_posi, inputs_nega): def process(self, pipe: WanVideoPipeline, inputs_shared, inputs_posi, inputs_nega):
if inputs_shared.get("animate_face_video", None) is None: if inputs_shared.get("animate_face_video", None) is None:
return {} return inputs_shared, inputs_posi, inputs_nega
inputs_posi["face_pixel_values"] = pipe.preprocess_video(inputs_shared["animate_face_video"]) inputs_posi["face_pixel_values"] = pipe.preprocess_video(inputs_shared["animate_face_video"])
inputs_nega["face_pixel_values"] = torch.zeros_like(inputs_posi["face_pixel_values"]) - 1 inputs_nega["face_pixel_values"] = torch.zeros_like(inputs_posi["face_pixel_values"]) - 1
return inputs_shared, inputs_posi, inputs_nega return inputs_shared, inputs_posi, inputs_nega
@@ -1374,6 +1374,7 @@ def model_fn_wan_video(
x = dit.patchify(x, control_camera_latents_input) x = dit.patchify(x, control_camera_latents_input)
# Animate # Animate
if pose_latents is not None and face_pixel_values is not None:
x, motion_vec = animate_adapter.after_patch_embedding(x, pose_latents, face_pixel_values) x, motion_vec = animate_adapter.after_patch_embedding(x, pose_latents, face_pixel_values)
# Patchify # Patchify