Update download-model.py

download-model.py

@@ -94,7 +94,7 @@ facebook/opt-1.3b
 
 def get_download_links_from_huggingface(model, branch):
     base = "https://huggingface.co"
-    page = f"/api/models/{model}/tree/{branch}?cursor="
+    page = f"/api/models/{model}/tree/{branch}"
     cursor = b""
 
     links = []
@@ -102,8 +102,11 @@ def get_download_links_from_huggingface(model, branch):
     has_pytorch = False
     has_safetensors = False
     while True:
-        content = requests.get(f"{base}{page}{cursor.decode()}").content
-
+        url = f"{base}{page}" + (f"?cursor={cursor.decode()}" if cursor else "")
+        r = requests.get(url)
+        r.raise_for_status()
+        content = r.content
+        
         dict = json.loads(content)
         if len(dict) == 0:
             break