Add debug logging for FlashVSR SegmentUpscale output shapes

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:
2026-02-13 16:31:09 +01:00
parent 11e2acb9e0
commit 39d0f7af42

View File

@@ -1895,6 +1895,8 @@ class FlashVSRSegmentUpscale:
from .flashvsr_arch.models.utils import 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
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
@@ -1921,4 +1923,5 @@ class FlashVSRSegmentUpscale:
else:
model._overlap_tail = None
logger.info(f"[FlashVSR Segment {segment_index}] final output: {result.shape}, dtype={result.dtype}, device={result.device}")
return (result, model)