Spaces:
Runtime error
Runtime error
algomuffin
commited on
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 |
|