fix flake8

Signed-off-by: Yiheng Wang <[email protected]>

Author: yiheng-wang-nv

generation/image_to_image_translation/tutorial_segmentation_with_ddpm.ipynb

@@ -741,7 +741,8 @@
                 "            )  # we only add noise to the segmentation mask\n",
                 "            combined = torch.cat(\n",
                 "                (images, noisy_seg), dim=1\n",
-                "            # we concatenate the brain MR image with the noisy segmenatation mask, to condition the generation process\n",
+                "            # we concatenate the brain MR image with the noisy segmenatation mask\n",
+                "            # to condition the generation process\n",
                 "            )\n",
                 "            prediction = model(x=combined, timesteps=timesteps)\n",
                 "            # Get model prediction\n",
@@ -1244,7 +1245,8 @@
                 "                    chain = torch.cat((chain, current_img.cpu()), dim=-1)\n",
                 "                combined = torch.cat(\n",
                 "                    (input_img, current_img), dim=1\n",
-                "                # in every step during the denoising process, the brain MR image is concatenated to add anatomical information\n",
+                "                # in every step during the denoising process\n",
+                "                # the brain MR image is concatenated to add anatomical information\n",
                 "                )\n",
                 "\n",
                 "    plt.style.use(\"default\")\n",