Spaces:

ChrisPreston
/

diff-svc_minato_aqua

Build error

ChrisPreston commited on Feb 8, 2023

Commit

fc9a60a

1 Parent(s): 830fb4c

Update modules/nsf_hifigan/models.py

Files changed (1) hide show

modules/nsf_hifigan/models.py CHANGED Viewed

@@ -25,7 +25,7 @@ def load_model(model_path, device='cuda'):
     generator = Generator(h).to(device)
-    cp_dict = torch.load(model_path)
     generator.load_state_dict(cp_dict['generator'])
     generator.eval()
     generator.remove_weight_norm()

     generator = Generator(h).to(device)
+    cp_dict = torch.load(model_path, map_location=torch.device('cpu'))
     generator.load_state_dict(cp_dict['generator'])
     generator.eval()
     generator.remove_weight_norm()