File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -441,8 +441,8 @@ def __call__(self, data):
441
441
442
442
if np .all (np .less (current_size , self .spatial_size )):
443
443
cropper = SpatialCrop (roi_center = center , roi_size = self .spatial_size )
444
- box_start = np . array ( [s .start for s in cropper .slices ])
445
- box_end = np . array ( [s .stop for s in cropper .slices ])
444
+ box_start = [s .start for s in cropper .slices ]
445
+ box_end = [s .stop for s in cropper .slices ]
446
446
else :
447
447
cropper = SpatialCrop (roi_start = box_start , roi_end = box_end )
448
448
Original file line number Diff line number Diff line change @@ -238,7 +238,7 @@ def step(
238
238
pred_prev_sample = pred_original_sample_coeff * pred_original_sample + current_sample_coeff * sample
239
239
240
240
# 6. Add noise
241
- variance = 0
241
+ variance : torch . Tensor | int = 0
242
242
if timestep > 0 :
243
243
noise = torch .randn (
244
244
model_output .size (),
You can’t perform that action at this time.
0 commit comments