Skip to content

Commit 44ecd9b

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 039e6fd commit 44ecd9b

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

generation/maisi/scripts/sample.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -333,7 +333,14 @@ def ldm_conditional_sample_one_image(
333333
latents, _ = noise_scheduler.step(model_output, t, latents, next_t) # type: ignore
334334
end_time = time.time()
335335
logging.info(f"---- DM/ControlNet Latent features generation time: {end_time - start_time} seconds ----")
336-
del unet_inputs, controlnet_inputs, model_output, controlnet_cond_vis, down_block_res_samples, mid_block_res_sample
336+
del (
337+
unet_inputs,
338+
controlnet_inputs,
339+
model_output,
340+
controlnet_cond_vis,
341+
down_block_res_samples,
342+
mid_block_res_sample,
343+
)
337344
gc.collect()
338345
torch.cuda.empty_cache()
339346

0 commit comments

Comments
 (0)