From 18ef72d7c0a3e3786bc7f4c135f69233ddcb6ca3 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sat, 21 Jan 2023 00:38:23 -0300
Subject: [PATCH] Update download-model.py

---
 download-model.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/download-model.py b/download-model.py
index 7b2e10c9..0519b18c 100644
--- a/download-model.py
+++ b/download-model.py
@@ -11,7 +11,6 @@ from bs4 import BeautifulSoup
 import multiprocessing
 import tqdm
 import sys
-from sys import argv
 import argparse
 from pathlib import Path
 import re
@@ -43,7 +42,7 @@ def sanitize_branch_name(branch_name):
         raise ValueError("Invalid branch name. Only alphanumeric characters, period, underscore and dash are allowed.")
 
 if __name__ == '__main__':
-    model = argv[1]
+    model = args.model
     if model[-1] == '/':
         model = model[:-1]
         branch = args.branch