Skip to content

Commit 9f4ae11

Browse files
committed
update doc
Signed-off-by: Can-Zhao <[email protected]>
1 parent 7c94a8d commit 9f4ae11

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

monai/networks/schedulers/rectified_flow.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -269,7 +269,7 @@ def step(
269269
next_timestep (Union[int, None]): Optional next timestep.
270270
271271
Returns:
272-
tuple[torch.Tensor, None]: Predicted sample at the next step and additional info.
272+
tuple[torch.Tensor, torch.Tensor]: Predicted sample at the next step and additional info.
273273
"""
274274
# Ensure num_inference_steps exists and is a valid integer
275275
if not hasattr(self, "num_inference_steps") or not isinstance(self.num_inference_steps, int):

0 commit comments

Comments
 (0)