fix: call check_and_upgrade_checkpoint()

EveryVoiceTTS · Nov 5, 2024 · fbf198b · fbf198b
1 parent 4badc36
commit fbf198b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/hfgl/utils.py b/hfgl/utils.py
@@ -42,9 +42,12 @@ def load_hifigan_from_checkpoint(ckpt: dict, device) -> Tuple[HiFiGAN, HiFiGANCo
         model = HiFiGAN(config).to(device)
     else:
         model = HiFiGANGenerator(config).to(device)
+
+    ckpt = model.check_and_upgrade_checkpoint(ckpt)
     model.load_state_dict(ckpt["state_dict"])
     model.generator.eval()
     model.generator.remove_weight_norm()
+
     return model, config