Spaces:
Runtime error
Runtime error
NimaBoscarino
commited on
Commit
β’
a67b67e
1
Parent(s):
27d3848
Remove the attempt_download call
Browse files- 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 |
|