reformat

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

Author: Can-Zhao

models/maisi_ct_generative/scripts/rectified_flow.py

@@ -12,7 +12,6 @@ def timestep_transform(
     t, input_img_size, base_img_size=32 * 32 * 32, scale=1.0, num_train_timesteps=1000, spatial_dim=3
 ):
     t = t / num_train_timesteps
-    resolution = input_img_size
     ratio_space = (input_img_size / base_img_size).pow(1.0 / spatial_dim)
 
     ratio = ratio_space * scale

models/maisi_ct_generative/scripts/sample.py

@@ -11,7 +11,6 @@
 
 import json
 import logging
-import math
 import os
 import random
 import time
@@ -20,7 +19,6 @@
 import monai
 import torch
 from monai.data import MetaTensor
-from monai.inferers import sliding_window_inference
 from monai.inferers.inferer import DiffusionInferer, SlidingWindowInferer
 from monai.transforms import Compose, SaveImage
 from monai.utils import set_determinism