Ethanfel/ComfyUI-Tween
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add debug logging for FlashVSR SegmentUpscale output shapes

Browse Source 
Helps diagnose issue where segment 1+ runs but produces no image output.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Ethanfel
2026-02-13 16:31:09 +01:00
parent 11e2acb9e0
commit 39d0f7af42
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
nodes.py
View File

@@ -1895,6 +1895,8 @@ class FlashVSRSegmentUpscale:
from .flashvsr_arch.models.utils import clean_vram from .flashvsr_arch.models.utils import clean_vram
clean_vram() clean_vram()
logger.info(f"[FlashVSR Segment {segment_index}] upscale result: {result.shape}, dtype={result.dtype}")
# Handle crossfade blending with previous segment's tail # Handle crossfade blending with previous segment's tail
if segment_index > 0 and overlap_frames > 0 and hasattr(model, '_overlap_tail'): if segment_index > 0 and overlap_frames > 0 and hasattr(model, '_overlap_tail'):
prev_tail = model._overlap_tail # [blend_frames, H, W, C] on CPU prev_tail = model._overlap_tail # [blend_frames, H, W, C] on CPU
@@ -1921,4 +1923,5 @@ class FlashVSRSegmentUpscale:
else: else:
model._overlap_tail = None model._overlap_tail = None
logger.info(f"[FlashVSR Segment {segment_index}] final output: {result.shape}, dtype={result.dtype}, device={result.device}")
return (result, model) return (result, model)