Update inference.ipynb

inference.ipynb

@@ -38,7 +38,22 @@
         "cellView": "form",
         "id": "noaoheUjvGbd"
       },
-      "execution_count": 1,
+      "execution_count": null,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "#@title [Run this if using Nvidia Ampere or newer. This will significantly speed up the process]\n",
+        "import torch\n",
+        "torch.backends.cuda.matmul.allow_tf32 = True\n",
+        "torch.backends.cudnn.allow_tf32 = True"
+      ],
+      "metadata": {
+        "cellView": "form",
+        "id": "MkGgqW87eUsQ"
+      },
+      "execution_count": null,
       "outputs": []
     },
     {
@@ -106,7 +121,7 @@
         "cellView": "form",
         "id": "VlPgrJf4wpHo"
       },
-      "execution_count": 4,
+      "execution_count": null,
       "outputs": []
     },
     {