From 5aed18d658c14d74e439b658b3b4dab865f3122f Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 14 May 2025 11:27:28 +0300
Subject: [PATCH] remove print

---
 wanvideo/utils/basic_flowmatch.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/wanvideo/utils/basic_flowmatch.py b/wanvideo/utils/basic_flowmatch.py
index 2c921ea..591510b 100644
--- a/wanvideo/utils/basic_flowmatch.py
+++ b/wanvideo/utils/basic_flowmatch.py
@@ -46,7 +46,6 @@ class FlowMatchScheduler():
         self.timesteps = self.timesteps.to(model_output.device)
         timestep_id = torch.argmin(
             (self.timesteps - timestep).abs(), dim=0)
-        print("timestep_id: ", timestep_id)
         sigma = self.sigmas[timestep_id].reshape(-1, 1, 1, 1)
         if to_final or (timestep_id + 1 >= len(self.timesteps)).any():
             sigma_ = 1 if (