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