Spaces:
Runtime error
Runtime error
Commit
·
cc182cb
1
Parent(s):
092a8ea
Try adding initial load to CPU to see if runtime error goes away
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ import open_clip
|
|
27 |
|
28 |
def load_model_from_config(config, ckpt, verbose=False):
|
29 |
print(f"Loading model from {ckpt}")
|
30 |
-
pl_sd = torch.load(ckpt, map_location="
|
31 |
sd = pl_sd["state_dict"]
|
32 |
model = instantiate_from_config(config.model)
|
33 |
m, u = model.load_state_dict(sd, strict=False)
|
|
|
27 |
|
28 |
def load_model_from_config(config, ckpt, verbose=False):
|
29 |
print(f"Loading model from {ckpt}")
|
30 |
+
pl_sd = torch.load(ckpt, map_location="cpu")
|
31 |
sd = pl_sd["state_dict"]
|
32 |
model = instantiate_from_config(config.model)
|
33 |
m, u = model.load_state_dict(sd, strict=False)
|