Simplify version check and make gdrive loading optional

LLaMA_LoRA.ipynb

@@ -60,20 +60,15 @@
         "# @title A small workaround { display-mode: \"form\" }\n",
         "# @markdown Don't panic if you see an error here. Just click the `RESTART RUNTIME` button in the output below, then Run All again.\n",
         "# @markdown The error will disappear on the next run.\n",
-        "!pip install Pillow==9.3.0 numpy==1.23.5\n",
+        "%pip install Pillow==9.3.0 numpy==1.23.5\n",
         "\n",
+        "import pkg_resources as r\n",
         "import PIL\n",
-        "major, minor = map(float, PIL.__version__.split(\".\")[:2])\n",
-        "version_float = major + minor / 10**len(str(minor))\n",
-        "print('PIL', version_float)\n",
-        "if version_float < 9.003:\n",
-        "    raise Exception(\"Restart the runtime by clicking the 'RESTART RUNTIME' button above (or Runtime > Restart Runtime).\")\n",
-        "\n",
         "import numpy\n",
-        "major, minor = map(float, numpy.__version__.split(\".\")[:2])\n",
-        "version_float = major + minor / 10**len(str(minor))\n",
-        "print('numpy', version_float)\n",
-        "if version_float < 1.0023:\n",
+        "for module, min_version in [(PIL, \"9.3\"), (numpy, \"1.23\")]:\n",
+        "  lib_version = r.parse_version(module.__version__)\n",
+        "  print(module.__name__, lib_version)\n",
+        "  if lib_version < r.parse_version(min_version):\n",
         "    raise Exception(\"Restart the runtime by clicking the 'RESTART RUNTIME' button above (or Runtime > Restart Runtime).\")"
       ],
       "metadata": {
@@ -144,15 +139,17 @@
         "# colab_notebook_name = remove_ipynb_extension(colab_notebook_filename)\n",
         "\n",
         "from google.colab import drive\n",
-        "drive.mount(google_drive_mount_path)\n",
+        "try:\n",
+        "  drive.mount(google_drive_mount_path)\n",
         "\n",
-        "# google_drive_data_directory_relative_path = f\"{google_drive_colab_data_folder}/{colab_notebook_name}\"\n",
-        "google_drive_data_directory_relative_path = google_drive_folder\n",
-        "google_drive_data_directory_path = f\"{google_drive_mount_path}/My Drive/{google_drive_data_directory_relative_path}\"\n",
-        "!mkdir -p \"{google_drive_data_directory_path}\"\n",
-        "!ln -nsf \"{google_drive_data_directory_path}\" ./data\n",
-        "!touch \"data/This folder is used by the Colab notebook \\\"{colab_notebook_filename}\\\".txt\"\n",
-        "!echo \"Data will be stored in Google Drive folder: \\\"{google_drive_data_directory_relative_path}\\\", which is mounted under \\\"{google_drive_data_directory_path}\\\"\"\n"
+        "  google_drive_data_directory_relative_path = google_drive_folder\n",
+        "  google_drive_data_directory_path = f\"{google_drive_mount_path}/My Drive/{google_drive_data_directory_relative_path}\"\n",
+        "  !mkdir -p \"{google_drive_data_directory_path}\"\n",
+        "  !ln -nsf \"{google_drive_data_directory_path}\" ./data\n",
+        "  !touch \"data/This folder is used by the Colab notebook \\\"{colab_notebook_filename}\\\".txt\"\n",
+        "  !echo \"Data will be stored in Google Drive folder: \\\"{google_drive_data_directory_relative_path}\\\", which is mounted under \\\"{google_drive_data_directory_path}\\\"\"\n",
+        "except Exception as e:\n",
+        "  print(\"Drive won't be mounted!\")"
       ],
       "metadata": {
         "id": "iZmRtUY68U5f"