We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7c94a8d commit 9f4ae11Copy full SHA for 9f4ae11
monai/networks/schedulers/rectified_flow.py
@@ -269,7 +269,7 @@ def step(
269
next_timestep (Union[int, None]): Optional next timestep.
270
271
Returns:
272
- tuple[torch.Tensor, None]: Predicted sample at the next step and additional info.
+ tuple[torch.Tensor, torch.Tensor]: Predicted sample at the next step and additional info.
273
"""
274
# Ensure num_inference_steps exists and is a valid integer
275
if not hasattr(self, "num_inference_steps") or not isinstance(self.num_inference_steps, int):
0 commit comments