Hugging Face's logo

Spaces:
Awesimo
/
jojogan
Runtime error

App Files Community
sztanki commited on
Commit
84e27df
·
1 Parent(s): 8b9a7d7

Reverse code

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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
- #generatorzombie.load_state_dict(ckptzombie["g"], strict=False)
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")