Skip to content

Commit 84e917f

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 7b71a61 commit 84e917f

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

monai/networks/schedulers/ddpm.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,6 @@
3131

3232
from __future__ import annotations
3333

34-
import numpy as np
3534
import torch
3635

3736
from monai.utils import StrEnum
@@ -123,7 +122,7 @@ def set_timesteps(self, num_inference_steps: int, device: str | torch.device | N
123122

124123
self.num_inference_steps = num_inference_steps
125124
self.timesteps = torch.linspace(self.num_train_timesteps - 1, 0, self.num_inference_steps, device=device).round().long()
126-
125+
127126

128127
def _get_mean(self, timestep: int, x_0: torch.Tensor, x_t: torch.Tensor) -> torch.Tensor:
129128
"""

0 commit comments

Comments
 (0)