Explorar el Código

Add generator only weight

Lengyue hace 2 años
padre
commit
ba1f1da9ff
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  1. 1 1
      fish_speech/models/vqgan/modules/firefly.py

+ 1 - 1
fish_speech/models/vqgan/modules/firefly.py

@@ -503,7 +503,7 @@ class FireflyBase(nn.Module):
             self.load_state_dict(torch.load(ckpt_path, map_location="cpu"))
             self.load_state_dict(torch.load(ckpt_path, map_location="cpu"))
         elif pretrained:
         elif pretrained:
             state_dict = torch.hub.load_state_dict_from_url(
             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",
                 map_location="cpu",
             )
             )