AmineSam commited on
Commit
d940fec
·
verified ·
1 Parent(s): 4222b56

Update models/experimental.py

Browse files
Files changed (1) hide show
  1. models/experimental.py +1 -1
models/experimental.py CHANGED
@@ -85,7 +85,7 @@ def attempt_load(weights, map_location=None):
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
 
91
  # Compatibility updates
 
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, weights_only=False) # load
89
  model.append(ckpt['ema' if ckpt.get('ema') else 'model'].float().fuse().eval()) # FP32 model
90
 
91
  # Compatibility updates