black

Can-Zhao · Can-Zhao · commit 39186fcd202a · 2025-03-05T21:31:36.000Z
Signed-off-by: Can-Zhao &lt;volcanofly@gmail.com&gt;
diff --git a/monai/networks/schedulers/rectified_flow.py b/monai/networks/schedulers/rectified_flow.py
@@ -291,6 +291,6 @@ def step(
             )  # Avoid division by zero
 
         pred_post_sample = sample + v_pred * dt
-        pred_original_sample = sample + v_pred * timestep/self.num_train_timesteps
-        
+        pred_original_sample = sample + v_pred * timestep / self.num_train_timesteps
+
         return pred_post_sample, pred_original_sample
