nodes/kijai-ComfyUI-WanVideoWrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

FlashVSR tweaks

Browse Source
This commit is contained in:
kijai
2025-10-16 00:42:33 +03:00
parent 367c612ddb
commit 50483a3b54
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
FlashVSR/TCDecoder.py
+1 -1
View File
@@ -241,7 +241,7 @@ class TAEHV(nn.Module):
trim_flag = self.mem[-8] is None # keeps original relative check trim_flag = self.mem[-8] is None # keeps original relative check
if cond is not None: if cond is not None:
shuffled = self.pixel_shuffle(cond) shuffled = self.pixel_shuffle(cond.to(x))
x = torch.cat([shuffled[:, :x.shape[1]], x], dim=2) x = torch.cat([shuffled[:, :x.shape[1]], x], dim=2)
x, self.mem = apply_model_with_memblocks(self.decoder, x, parallel, show_progress_bar, mem=self.mem) x, self.mem = apply_model_with_memblocks(self.decoder, x, parallel, show_progress_bar, mem=self.mem)
example_workflows/wanvideo_1_3B_FlashVSR_upscale_example.json
+1 -1
View File
@@ -875,7 +875,7 @@
"Node name for S&R": "WanVideoExtraModelSelect" "Node name for S&R": "WanVideoExtraModelSelect"
}, },
"widgets_values": [ "widgets_values": [
"WanVideo\\FlashVSR\\Wan2_1_FlashVSR_PQ_proj_model_bf16.safetensors" "WanVideo\\FlashVSR\\Wan2_1_FlashVSR_LQ_proj_model_bf16.safetensors"
], ],
"color": "#223", "color": "#223",
"bgcolor": "#335" "bgcolor": "#335"
nodes_sampler.py
+3 -3
View File
@@ -838,9 +838,9 @@ class WanVideoSampler:
missing_frames = num_frames + 4 - flashvsr_LQ_images.shape[0] missing_frames = num_frames + 4 - flashvsr_LQ_images.shape[0]
last_frame = flashvsr_LQ_images[-1:].repeat(missing_frames, 1, 1, 1) last_frame = flashvsr_LQ_images[-1:].repeat(missing_frames, 1, 1, 1)
flashvsr_LQ_images = torch.cat([flashvsr_LQ_images, last_frame], dim=0) flashvsr_LQ_images = torch.cat([flashvsr_LQ_images, last_frame], dim=0)
LQ_images = flashvsr_LQ_images[:num_frames+4].unsqueeze(0).movedim(-1, 1).to(device, dtype) * 2 - 1 LQ_images = flashvsr_LQ_images[:num_frames+4].unsqueeze(0).movedim(-1, 1).to(dtype) * 2 - 1
if context_options is None: if context_options is None:
flashvsr_LQ_latent = transformer.LQ_proj_in(LQ_images) flashvsr_LQ_latent = transformer.LQ_proj_in(LQ_images.to(device))
log.info(f"flashvsr_LQ_latent: {flashvsr_LQ_latent[0].shape}") log.info(f"flashvsr_LQ_latent: {flashvsr_LQ_latent[0].shape}")
seq_len = math.ceil((noise.shape[2] * noise.shape[3]) / 4 * noise.shape[1]) seq_len = math.ceil((noise.shape[2] * noise.shape[3]) / 4 * noise.shape[1])
@@ -1955,7 +1955,7 @@ class WanVideoSampler:
end = c[-1] * 4 + 1 + 4 end = c[-1] * 4 + 1 + 4
center_indices = torch.arange(start, end, 1) center_indices = torch.arange(start, end, 1)
center_indices = torch.clamp(center_indices, min=0, max=LQ_images.shape[2] - 1) center_indices = torch.clamp(center_indices, min=0, max=LQ_images.shape[2] - 1)
partial_flashvsr_LQ_images = LQ_images[:, :, center_indices].to(device, dtype) partial_flashvsr_LQ_images = LQ_images[:, :, center_indices].to(device)
partial_flashvsr_LQ_latent = transformer.LQ_proj_in(partial_flashvsr_LQ_images) partial_flashvsr_LQ_latent = transformer.LQ_proj_in(partial_flashvsr_LQ_images)
if len(timestep.shape) != 1: if len(timestep.shape) != 1: