Type fix?

Signed-off-by: Eric Kerfoot <[email protected]>

Author: ericspod

monai/networks/schedulers/ddpm.py

@@ -238,7 +238,7 @@ def step(
         pred_prev_sample = pred_original_sample_coeff * pred_original_sample + current_sample_coeff * sample
 
         # 6. Add noise
-        variance = 0
+        variance: int | torch.Tensor = 0
         if timestep > 0:
             noise = torch.randn(
                 model_output.size(),