glenn-jocher commited on
Commit
77fd832
1 Parent(s): 8dc68fc

Update google_utils.py (#2017)

Browse files
Files changed (1) hide show
  1. utils/google_utils.py +2 -2
utils/google_utils.py CHANGED
@@ -26,8 +26,8 @@ def attempt_download(file, repo='ultralytics/yolov5'):
26
  assets = [x['name'] for x in response['assets']] # release assets, i.e. ['yolov5s.pt', 'yolov5m.pt', ...]
27
  tag = response['tag_name'] # i.e. 'v1.0'
28
  except: # fallback plan
29
- assets = ['yolov5.pt', 'yolov5.pt', 'yolov5l.pt', 'yolov5x.pt']
30
- tag = subprocess.check_output('git tag', shell=True).decode('utf-8').split('\n')[-2]
31
 
32
  name = file.name
33
  if name in assets:
 
26
  assets = [x['name'] for x in response['assets']] # release assets, i.e. ['yolov5s.pt', 'yolov5m.pt', ...]
27
  tag = response['tag_name'] # i.e. 'v1.0'
28
  except: # fallback plan
29
+ assets = ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt']
30
+ tag = subprocess.check_output('git tag', shell=True).decode().split()[-1]
31
 
32
  name = file.name
33
  if name in assets: