Skip to content

Commit bbce5fc

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 306f3e7 commit bbce5fc

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

generation/realism_diversity_metrics/realism_diversity_metrics.ipynb

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -262,7 +262,6 @@
262262
"metadata": {},
263263
"outputs": [],
264264
"source": [
265-
"\n",
266265
"import torch\n",
267266
"\n",
268267
"unet_url = \"https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/tutorial_generation_unet.pth\"\n",
@@ -327,7 +326,9 @@
327326
"scheduler.set_timesteps(num_inference_steps=1000)\n",
328327
"\n",
329328
"with torch.no_grad():\n",
330-
" syn_images = inferer.sample(input_noise=noise, diffusion_model=unet, scheduler=scheduler, autoencoder_model=autoencoderkl)\n",
329+
" syn_images = inferer.sample(\n",
330+
" input_noise=noise, diffusion_model=unet, scheduler=scheduler, autoencoder_model=autoencoderkl\n",
331+
" )\n",
331332
"\n",
332333
"# Plot 3 examples from the synthetic data\n",
333334
"fig, ax = plt.subplots(nrows=1, ncols=3)\n",
@@ -395,7 +396,9 @@
395396
" scheduler.set_timesteps(num_inference_steps=1000)\n",
396397
"\n",
397398
" with torch.no_grad():\n",
398-
" syn_images = inferer.sample(input_noise=noise, diffusion_model=unet, scheduler=scheduler, autoencoder_model=autoencoderkl)\n",
399+
" syn_images = inferer.sample(\n",
400+
" input_noise=noise, diffusion_model=unet, scheduler=scheduler, autoencoder_model=autoencoderkl\n",
401+
" )\n",
399402
"\n",
400403
" # Get the features for the real data\n",
401404
" real_eval_feats = get_features(real_images)\n",

0 commit comments

Comments
 (0)