From 38897fbd8a88472aadb4122c2a07b75e9abe3958 Mon Sep 17 00:00:00 2001
From: Roman <Romanthefrog@gmail.com>
Date: Sat, 24 Jun 2023 14:09:34 +0100
Subject: [PATCH] fix: added model parameter check (#2829)

---
 download-model.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/download-model.py b/download-model.py
index e04aedc3..87fd6261 100644
--- a/download-model.py
+++ b/download-model.py
@@ -217,6 +217,10 @@ if __name__ == '__main__':
     branch = args.branch
     model = args.MODEL
 
+    if model is None:
+        print("Error: Please specify the model you'd like to download (e.g. 'python download-model.py facebook/opt-1.3b').")
+        sys.exit()
+
     downloader = ModelDownloader()
     # Cleaning up the model/branch names
     try: