Reverse code
Browse files
app.py
CHANGED
@@ -82,8 +82,7 @@ transform = transforms.Compose(
|
|
82 |
#ZOMBIE
|
83 |
modelzombie = hf_hub_download(repo_id="Awesimo/jojogan-zombie", filename="zombie.pt")
|
84 |
ckptzombie = torch.load(modelzombie, map_location=lambda storage, loc: storage)
|
85 |
-
|
86 |
-
generatorzombie.load_state_dict(ckptzombie, strict=False)
|
87 |
|
88 |
#JOJO
|
89 |
modeljojo = hf_hub_download(repo_id="akhaliq/JoJoGAN-jojo", filename="jojo_preserve_color.pt")
|
|
|
82 |
#ZOMBIE
|
83 |
modelzombie = hf_hub_download(repo_id="Awesimo/jojogan-zombie", filename="zombie.pt")
|
84 |
ckptzombie = torch.load(modelzombie, map_location=lambda storage, loc: storage)
|
85 |
+
generatorzombie.load_state_dict(ckptzombie["g"], strict=False)
|
|
|
86 |
|
87 |
#JOJO
|
88 |
modeljojo = hf_hub_download(repo_id="akhaliq/JoJoGAN-jojo", filename="jojo_preserve_color.pt")
|