fix: f5tts rename #267

Author: zhzLuke96

modules/core/models/tts/F5/F5ttsApi.py

@@ -29,8 +29,10 @@
 
 # NOTE: 目前不支持 bigvgan 因为需要引入外部库，并且似乎没什么特别区别
 class F5TTS:
+
     def __init__(
         self,
+        ckpt_file: Path,
         model: Literal["F5TTS_v1_Base", "F5TTS_Base"] = "F5TTS_v1_Base",
         ode_method="euler",
         use_ema=True,
@@ -47,13 +49,6 @@ def __init__(
             / "configs"
             / f"{model}.yaml"
         )
-        ckpt_file = (
-            Path(os.getcwd())
-            / "models"
-            / "F5-TTS"
-            / model
-            / "model_1200000.safetensors"
-        )
         vocab_file = (
             Path(os.getcwd())
             / "modules"
@@ -64,16 +59,6 @@ def __init__(
             / "Emilia_ZH_EN_pinyin"
             / f"vocab.txt"
         )
-        if not ckpt_file.exists():
-            # NOTE: 不存在，提示使用脚本下载 `python -m scripts.downloader.f5_tts_v1 --source huggingface`
-            if model == "F5TTS_Base":
-                raise ValueError(
-                    f"F5TTS model {model} not found, please download it manually using `python -m scripts.downloader.f5_tts --source huggingface`"
-                )
-            else:
-                raise ValueError(
-                    f"F5TTS model {model} not found, please download it manually using `python -m scripts.downloader.f5_tts_v1 --source huggingface`"
-                )
 
         # Path to str path, 因为f5的函数不支持Path
         ckpt_file = str(ckpt_file)

modules/core/models/tts/F5TtsModel.py

@@ -23,7 +23,16 @@ def __init__(self) -> None:
         super().__init__("f5-tts")
 
         v0_6_model_path = Path("./models/F5-TTS/F5TTS_Base/model_1200000.safetensors")
-        v1_model_path = Path("./models/F5-TTS/F5TTS_v1_Base/model_1200000.safetensors")
+        v1_120_model_path = Path(
+            "./models/F5-TTS/F5TTS_v1_Base/model_1200000.safetensors"
+        )
+        # NOTE: F5TTS 在 2025三月份发布的新 v1_base 模型
+        v1_125_model_path = Path(
+            "./models/F5-TTS/F5TTS_v1_Base/model_1250000.safetensors"
+        )
+        v1_model_path = (
+            v1_125_model_path if v1_125_model_path.exists() else v1_120_model_path
+        )
 
         self.model_path = v1_model_path
         self.model_version = "F5TTS_v1_Base"
@@ -55,12 +64,29 @@ def check_files(self) -> None:
     def is_downloaded(self) -> bool:
         return self.model_path.exists() and self.vocos_path.exists()
 
+    def _alert_ckpt_files(self):
+        """
+        提醒下载模型
+        """
+        if not self.model_path.exists():
+            # NOTE: 不存在，提示使用脚本下载 `python -m scripts.downloader.f5_tts_v1 --source huggingface`
+            if self.model_version == "F5TTS_Base":
+                raise ValueError(
+                    f"F5TTS model {self.model_version} not found, please download it manually using `python -m scripts.downloader.f5_tts --source huggingface`"
+                )
+            else:
+                raise ValueError(
+                    f"F5TTS model {self.model_version} not found, please download it manually using `python -m scripts.downloader.f5_tts_v1 --source huggingface`"
+                )
+
     def load(self) -> F5TTS:
+        self._alert_ckpt_files()
         self.check_files()
 
         with self.load_lock:
             if self.model is None:
                 self.model = F5TTS(
+                    ckpt_file=self.model_path,
                     model=self.model_version,
                     vocoder_local_path=str(self.vocos_path),
                     device=self.get_device(),

scripts/downloader/f5_tts.py

@@ -31,6 +31,7 @@ def __init__(self):
             modelscope_repo="AI-ModelScope/F5-TTS",
             huggingface_repo="SWivid/F5-TTS",
             required_files=required_files,
+            just_download_required_files=True,
         )
 
         self.logger = logger

scripts/downloader/f5_tts_v1.py

@@ -31,6 +31,7 @@ def __init__(self):
             modelscope_repo="AI-ModelScope/F5-TTS",
             huggingface_repo="SWivid/F5-TTS",
             required_files=required_files,
+            just_download_required_files=True,
         )
 
         self.logger = logger
@@ -60,7 +61,7 @@ def from_modelscope(self):
 
         # v1 版本是 2025/3 发布的新模型
         url = "https://modelscope.cn/models/AI-ModelScope/F5-TTS/resolve/master/F5TTS_v1_Base/model_1250000.safetensors"
-        dest_path = self.model_dir / "F5TTS_v1_Base" / "model_1200000.safetensors"
+        dest_path = self.model_dir / "F5TTS_v1_Base" / "model_1250000.safetensors"
 
         self._download_with_progress(url, dest_path)