remove print

2025-05-14 11:27:28 +03:00
parent 1701375214
commit 5aed18d658
1 changed files with 0 additions and 1 deletions
@@ -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 (