Spaces:
Runtime error
Runtime error
Commit
·
e758fa3
1
Parent(s):
5e4f4cb
Update app.py
Browse files
app.py
CHANGED
|
@@ -165,8 +165,8 @@ generatorsketch.load_state_dict(ckptsketch["g"], strict=False)
|
|
| 165 |
|
| 166 |
modeldisneynoc = hf_hub_download(repo_id="algomuffin/disney", filename="disney.pt")
|
| 167 |
|
| 168 |
-
|
| 169 |
-
|
| 170 |
|
| 171 |
|
| 172 |
|
|
|
|
| 165 |
|
| 166 |
modeldisneynoc = hf_hub_download(repo_id="algomuffin/disney", filename="disney.pt")
|
| 167 |
|
| 168 |
+
ckptMy = torch.load(modeldisneynoc, map_location=lambda storage, loc: storage)
|
| 169 |
+
generatordisneynoc.load_state_dict(ckptMy["g"], strict=False)
|
| 170 |
|
| 171 |
|
| 172 |
|