From f18eda88c2b39a9aab73bd8fb61ae5bb8d29e4f3 Mon Sep 17 00:00:00 2001
From: zetavg <mail@zeta.vg>
Date: Fri, 14 Apr 2023 21:25:01 +0800
Subject: [PATCH] resolves #2 numpy version mismatch on colab

---
 LLaMA_LoRA.ipynb | 12 ++++++++++--
 README.md        |  5 +++++
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/LLaMA_LoRA.ipynb b/LLaMA_LoRA.ipynb
index 230a5c4..a5bd136 100644
--- a/LLaMA_LoRA.ipynb
+++ b/LLaMA_LoRA.ipynb
@@ -60,11 +60,19 @@
         "# @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\n",
+        "!pip install Pillow==9.3.0 numpy==1.23.5\n",
+        "\n",
         "import PIL\n",
         "major, minor = map(float, PIL.__version__.split(\".\")[:2])\n",
         "version_float = major + minor / 10**len(str(minor))\n",
-        "print(version_float)\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 < 9.003:\n",
         "    raise Exception(\"Restart the runtime by clicking the 'RESTART RUNTIME' button above (or Runtime > Restart Runtime).\")"
       ],
diff --git a/README.md b/README.md
index 6fa230e..7ad4ea3 100644
--- a/README.md
+++ b/README.md
@@ -114,6 +114,11 @@ For more options, see `python app.py --help`.
 </details>
 
 
+## Usage
+
+See [video on YouTube](https://youtu.be/IoEMgouZ5xU).
+
+
 ## Acknowledgements
 
 * https://github.com/tloen/alpaca-lora