feat: fixed return type annotation

EveryVoiceTTS · Nov 26, 2024 · 73e144f · 73e144f
1 parent 38f946d
commit 73e144f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/hfgl/utils.py b/hfgl/utils.py
@@ -21,7 +21,9 @@ def sizeof_fmt(num, suffix="B"):
     return f"{num:.1f}Yi{suffix}"
 
 
-def load_hifigan_from_checkpoint(ckpt: dict, device) -> Tuple[HiFiGAN, HiFiGANConfig]:
+def load_hifigan_from_checkpoint(
+    ckpt: dict, device
+) -> Tuple[HiFiGAN | HiFiGANGenerator, HiFiGANConfig]:
     config: dict | HiFiGANConfig = ckpt["hyper_parameters"]["config"]
     if isinstance(config, dict):
         from pydantic import ValidationError