Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-11-04 14:16:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix autoresize Hunyuan Avatar ref image

Browse Source
This commit is contained in:
DeepBeepMeep 2025-06-06 13:40:09 +02:00
parent 5c850fb29e
commit eae83e8acc
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
hyvideo/hunyuan.py
View File

@ -867,6 +867,9 @@ class HunyuanVideoSampler(Inference):
if self.avatar: if self.avatar:
w, h = input_ref_images.size w, h = input_ref_images.size
target_height, target_width = calculate_new_dimensions(target_height, target_width, h, w, fit_into_canvas) target_height, target_width = calculate_new_dimensions(target_height, target_width, h, w, fit_into_canvas)
if target_width != w or target_height != h:
input_ref_images = input_ref_images.resize((target_width,target_height), resample=Image.Resampling.LANCZOS)
concat_dict = {'mode': 'timecat', 'bias': -1} concat_dict = {'mode': 'timecat', 'bias': -1}
freqs_cos, freqs_sin = self.get_rotary_pos_embed_new(129, target_height, target_width, concat_dict) freqs_cos, freqs_sin = self.get_rotary_pos_embed_new(129, target_height, target_width, concat_dict)
else: else:

6
wgp.py
View File

@ -3263,13 +3263,13 @@ def generate_video(
if exp > 0: if exp > 0:
from rife.inference import temporal_interpolation from rife.inference import temporal_interpolation
if sliding_window and window_no > 1: if sliding_window and window_no > 1:
sample = torch.cat([previous_before_last_frame, sample], dim=1) sample = torch.cat([previous_last_frame, sample], dim=1)
previous_before_last_frame = sample[:, -2:-1].clone() previous_last_frame = sample[:, -1:].clone()
sample = temporal_interpolation( os.path.join("ckpts", "flownet.pkl"), sample, exp, device=processing_device) sample = temporal_interpolation( os.path.join("ckpts", "flownet.pkl"), sample, exp, device=processing_device)
sample = sample[:, 1:] sample = sample[:, 1:]
else: else:
sample = temporal_interpolation( os.path.join("ckpts", "flownet.pkl"), sample, exp, device=processing_device) sample = temporal_interpolation( os.path.join("ckpts", "flownet.pkl"), sample, exp, device=processing_device)
previous_before_last_frame = sample[:, -2:-1].clone() previous_last_frame = sample[:, -1:].clone()
output_fps = output_fps * 2**exp output_fps = output_fps * 2**exp