Skip to content

Commit 8167f42

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 3fd54f2 commit 8167f42

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

generation/maisi/maisi_inference_tutorial.ipynb

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -796,11 +796,15 @@
796796
"controlnet.load_state_dict(checkpoint_controlnet[\"controlnet_state_dict\"], strict=True)\n",
797797
"\n",
798798
"mask_generation_autoencoder = define_instance(args, \"mask_generation_autoencoder_def\").to(device)\n",
799-
"checkpoint_mask_generation_autoencoder = torch.load(args.trained_mask_generation_autoencoder_path, weights_only=True, map_location=device)\n",
799+
"checkpoint_mask_generation_autoencoder = torch.load(\n",
800+
" args.trained_mask_generation_autoencoder_path, weights_only=True, map_location=device\n",
801+
")\n",
800802
"mask_generation_autoencoder.load_state_dict(checkpoint_mask_generation_autoencoder)\n",
801803
"\n",
802804
"mask_generation_diffusion_unet = define_instance(args, \"mask_generation_diffusion_def\").to(device)\n",
803-
"checkpoint_mask_generation_diffusion_unet = torch.load(args.trained_mask_generation_diffusion_path, weights_only=True, map_location=device)\n",
805+
"checkpoint_mask_generation_diffusion_unet = torch.load(\n",
806+
" args.trained_mask_generation_diffusion_path, weights_only=True, map_location=device\n",
807+
")\n",
804808
"mask_generation_diffusion_unet.load_state_dict(checkpoint_mask_generation_diffusion_unet[\"unet_state_dict\"])\n",
805809
"mask_generation_scale_factor = checkpoint_mask_generation_diffusion_unet[\"scale_factor\"]\n",
806810
"\n",

0 commit comments

Comments
 (0)