fixed the problem of downloading models (WongKinYiu#451)

mxm1994 · Aug 8, 2022 · c14ba0c · c14ba0c
1 parent 8758cdd
commit c14ba0c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/utils/google_utils.py b/utils/google_utils.py
@@ -26,7 +26,8 @@ def attempt_download(file, repo='WongKinYiu/yolov7'):
             assets = [x['name'] for x in response['assets']]  # release assets
             tag = response['tag_name']  # i.e. 'v1.0'
         except:  # fallback plan
-            assets = ['yolov7.pt']
+            assets = ['yolov7.pt', 'yolov7-tiny.pt', 'yolov7x.pt', 'yolov7-d6.pt', 'yolov7-e6.pt', 
+                      'yolov7-e6e.pt', 'yolov7-w6.pt']
             tag = subprocess.check_output('git tag', shell=True).decode().split()[-1]
 
         name = file.name