Don't zero the extra frames
Possible fix for SVI "shot" method
This commit is contained in:
@@ -922,7 +922,7 @@ class WanVideoImageToVideoEncode:
|
||||
del resized_start_image, zero_frames
|
||||
else:
|
||||
temporal_mask = common_upscale(temporal_mask.unsqueeze(1), W, H, "nearest", "disabled").squeeze(1)
|
||||
concatenated = resized_start_image[:,:num_frames].to(vae.dtype) * temporal_mask[:num_frames].unsqueeze(0).to(vae.dtype)
|
||||
concatenated = resized_start_image[:,:num_frames].to(vae.dtype)# * temporal_mask[:num_frames].unsqueeze(0).to(vae.dtype)
|
||||
del resized_start_image, temporal_mask
|
||||
|
||||
mm.soft_empty_cache()
|
||||
|
||||
Reference in New Issue
Block a user