make it 2D/3D compartible, rm a outdated comment

Signed-off-by: Can-Zhao <[email protected]>

Author: Can-Zhao

monai/networks/schedulers/rectified_flow.py

@@ -179,7 +179,6 @@ def add_noise(self, original_samples: torch.Tensor, noise: torch.Tensor, timeste
         timepoints = 1 - timepoints  # [1,1/1000]
 
         # expand timepoint to noise shape
-        # Just in case timepoints is not 1D or 2D tensor, make it to be same shape as noise
         if len(noise.shape) == 5:
             timepoints = timepoints.unsqueeze(1).unsqueeze(1).unsqueeze(1).unsqueeze(1)
             timepoints = timepoints.repeat(1, noise.shape[1], noise.shape[2], noise.shape[3], noise.shape[4])