From bb66a4d32c773299f2e5ffeb05a35f97372dadb1 Mon Sep 17 00:00:00 2001 From: smthemex <138738845+smthemex@users.noreply.github.com> Date: Wed, 22 Oct 2025 20:39:25 +0800 Subject: [PATCH] Update flashvsr_full.py --- FlashVSR/diffsynth/pipelines/flashvsr_full.py | 33 ++++++++++--------- 1 file changed, 17 insertions(+), 16 deletions(-) diff --git a/FlashVSR/diffsynth/pipelines/flashvsr_full.py b/FlashVSR/diffsynth/pipelines/flashvsr_full.py index 46b6925..b261596 100644 --- a/FlashVSR/diffsynth/pipelines/flashvsr_full.py +++ b/FlashVSR/diffsynth/pipelines/flashvsr_full.py @@ -164,6 +164,7 @@ class FlashVSRFullPipeline(BasePipeline): self.prompt_emb_posi = None self.ColorCorrector = TorchColorCorrectorWavelet(levels=5) + print(r""" ███████╗██╗ █████╗ ███████╗██╗ ██╗██╗ ██╗███████╗█████╗ ██╔════╝██║ ██╔══██╗██╔════╝██║ ██║██║ ██║██╔════╝██╔══██╗ @@ -432,25 +433,25 @@ class FlashVSRFullPipeline(BasePipeline): latents_total.append(cur_latents) latents = torch.cat(latents_total, dim=2) - self.dit.to("cpu") - torch.cuda.empty_cache + #self.dit.to("cpu") + #torch.cuda.empty_cache # Decode - frames = self.decode_video(latents, **tiler_kwargs) + #frames = self.decode_video(latents, **tiler_kwargs) # 颜色校正（wavelet） - try: - if color_fix: - frames = self.ColorCorrector( - frames.to(device=LQ_video.device), - LQ_video[:, :, :frames.shape[2], :, :], - clip_range=(-1, 1), - chunk_size=16, - method='adain' - ) - except: - pass - - return frames[0] + # try: + # if color_fix: + # frames = self.ColorCorrector( + # frames.to(device=LQ_video.device), + # LQ_video[:, :, :frames.shape[2], :, :], + # clip_range=(-1, 1), + # chunk_size=16, + # method='adain' + # ) + # except: + # pass + + return latents #frames[0] # -----------------------------