|
|
@@ -503,7 +503,7 @@ class FireflyBase(nn.Module):
|
|
|
self.load_state_dict(torch.load(ckpt_path, map_location="cpu"))
|
|
|
elif pretrained:
|
|
|
state_dict = torch.hub.load_state_dict_from_url(
|
|
|
- "https://github.com/fishaudio/vocoder/releases/download/1.0.0/firefly-gan-base.ckpt",
|
|
|
+ "https://github.com/fishaudio/vocoder/releases/download/1.0.0/firefly-gan-base-generator.ckpt",
|
|
|
map_location="cpu",
|
|
|
)
|
|
|
|