From 017e672436cfc8d97f35cc1def369772eb89010a Mon Sep 17 00:00:00 2001
From: John Ramsden <johnramsden@riseup.net>
Date: Sun, 13 Oct 2024 16:33:00 -0700
Subject: [PATCH] Add docker-compose check

Running './run.sh' gives:

$ ./run.sh --model 7b
unknown shorthand flag: 'f' in -f

Add a check for docker compose being installed.
---
 run.sh | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/run.sh b/run.sh
index bd5b32e..7448232 100755
--- a/run.sh
+++ b/run.sh
@@ -1,12 +1,19 @@
 #!/bin/bash
 
-# Check if docker compose is installed
+# Check if docker is installed
 if ! command -v docker &> /dev/null
 then
     echo "Docker could not be found. Please install Docker and try again."
     exit
 fi
 
+# Check if docker composeis installed
+if ! docker compose version &> /dev/null
+then
+    echo "\"Docker compose\" could not be found. Please install \"Docker compose\" and try again."
+    exit
+fi
+
 # Parse command line arguments for model value and check for --with-cuda flag
 with_cuda=0
 while [[ "$#" -gt 0 ]]; do