Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -98,7 +98,7 @@ generatorgollum_mod.load_state_dict(ckptgollum_mod["g"], strict=False)
|
|
98 |
|
99 |
modelgollum_ex = hf_hub_download(repo_id="hlydecker/gandalf-gollum-extreme", filename="gollum_extreme.pt")
|
100 |
|
101 |
-
ckptgollum_ex = torch.load(
|
102 |
generatorgollum_ex.load_state_dict(ckptgollum_ex["g"], strict=False)
|
103 |
|
104 |
|
|
|
98 |
|
99 |
modelgollum_ex = hf_hub_download(repo_id="hlydecker/gandalf-gollum-extreme", filename="gollum_extreme.pt")
|
100 |
|
101 |
+
ckptgollum_ex = torch.load(modelgollum_ex, map_location=lambda storage, loc: storage)
|
102 |
generatorgollum_ex.load_state_dict(ckptgollum_ex["g"], strict=False)
|
103 |
|
104 |
|