fix ci

Signed-off-by: YunLiu <[email protected]>

Author: KumoLiu

generation/image_to_image_translation/tutorial_segmentation_with_ddpm.ipynb

@@ -582,7 +582,7 @@
     "    progress_bar = tqdm(scheduler.timesteps)\n",
     "    chain = torch.zeros(current_img.shape)\n",
     "    for t in progress_bar:  # go through the noising process\n",
-    "        with autocast(enabled=False):\n",
+    "        with autocast(\"cuda\", enabled=False):\n",
     "            with torch.no_grad():\n",
     "                model_output = model(combined, timesteps=torch.Tensor((t,)).to(current_img.device))\n",
     "                current_img, _ = scheduler.step(\n",