Create a data_dir on the drive if it doesn't exist

ddPn08 · Jan 14, 2023 · c2251f1 · c2251f1
1 parent 3fbdcf4
commit c2251f1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/automatic1111.ipynb b/automatic1111.ipynb
@@ -133,6 +133,7 @@
         "    try:\n",
         "        drive.mount(drive_path, force_remount=force_remount)\n",
         "        data_dir_gdrive = \"/content/drive/MyDrive/AI/automatic1111\"  # @param {type:\"string\"}\n",
+        "        os.makedirs(data_dir_gdrive, exist_ok=True)\n",
         "        ! rm -Rf {data_dir} && ln -s {data_dir_gdrive} {data_dir}\n",
         "        mount_success = True\n",
         "    except:\n",