NimaBoscarino commited on
Commit
a67b67e
β€’
1 Parent(s): 27d3848

Remove the attempt_download call

Browse files
Files changed (1) hide show
  1. models/experimental.py +1 -1
models/experimental.py CHANGED
@@ -84,7 +84,7 @@ def attempt_load(weights, map_location=None):
84
  # Loads an ensemble of models weights=[a,b,c] or a single model weights=[a] or weights=a
85
  model = Ensemble()
86
  for w in weights if isinstance(weights, list) else [weights]:
87
- attempt_download(w)
88
  ckpt = torch.load(w, map_location=map_location) # load
89
  model.append(ckpt['ema' if ckpt.get('ema') else 'model'].float().fuse().eval()) # FP32 model
90
 
84
  # Loads an ensemble of models weights=[a,b,c] or a single model weights=[a] or weights=a
85
  model = Ensemble()
86
  for w in weights if isinstance(weights, list) else [weights]:
87
+ # attempt_download(w)
88
  ckpt = torch.load(w, map_location=map_location) # load
89
  model.append(ckpt['ema' if ckpt.get('ema') else 'model'].float().fuse().eval()) # FP32 model
90