From: Jack Vial <redacted>
Date: Mon, 22 Jan 2024 12:03:48 +0000 (-0500)
Subject: mnist : add tensorflow and keras to requirements.txt (#702)
X-Git-Tag: upstream/0.0.1642~1054
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=ca7e136d7c4cd56399ea399b0757fb3806d91e80;p=pkg%2Fggml%2Fsources%2Fggml

mnist : add tensorflow and keras to requirements.txt (#702)

* Added tensorflow and keras to requirements.txt to streamline running MNIST example. Updated instructions to include installing python dependenices for the training script.

* add output example whitespace back

* edit readme and fixed typo
---

diff --git a/examples/mnist/README.md b/examples/mnist/README.md
index f6b66f74..32e949d4 100644
--- a/examples/mnist/README.md
+++ b/examples/mnist/README.md
@@ -3,11 +3,16 @@
 These are simple examples of how to use GGML for inferencing.
 The first example uses convolutional neural network (CNN), the second one uses fully connected neural network.
 
-## Building the examples
+## Python environment setup and build the examples
 
 ```bash
 git clone https://github.com/ggerganov/ggml
 cd ggml
+# Install python dependencies in a virtual environment
+python3 -m venv ggml_env
+source ./ggml_env/bin/activate
+pip install -r requirements.txt
+# Build the examples
 mkdir build && cd build
 cmake ..
 make -j4 mnist-cnn mnist
diff --git a/requirements.txt b/requirements.txt
index a9fd9604..c686d7fd 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,3 +6,5 @@ torchaudio==2.0.2
 torchvision==0.15.2
 transformers==4.29.2
 gguf==0.4.5
+tensorflow==2.15.0
+keras==2.15.0