fix ci

Signed-off-by: Yun Liu <yunl@nvidia.com>

Author: KumoLiu

generation/maisi/maisi_inference_tutorial.ipynb

@@ -366,7 +366,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": null,
    "id": "d499f7b1",
    "metadata": {
     "lines_to_next_cell": 2
@@ -391,7 +391,7 @@
     "noise_scheduler = define_instance(args, \"noise_scheduler\")\n",
     "mask_generation_noise_scheduler = define_instance(args, \"mask_generation_noise_scheduler\")\n",
     "\n",
-    "device = torch.device(\"cuda\")\n",
+    "device = torch.device(\"cuda\") if torch.cuda.is_available() else torch.device(\"cpu\")\n",
     "\n",
     "autoencoder = define_instance(args, \"autoencoder_def\").to(device)\n",
     "checkpoint_autoencoder = torch.load(args.trained_autoencoder_path, weights_only=True)\n",