From 2e4c601d1d671155632189ef9ea4ee0b9fd94a44 Mon Sep 17 00:00:00 2001 From: Pine Date: Wed, 6 May 2026 14:58:46 +0800 Subject: [PATCH] =?UTF-8?q?-=20=E4=BF=AE=E5=A4=8D=E8=B7=AF=E5=BE=84?= =?UTF-8?q?=E9=94=99=E8=AF=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app.py | 6 +++--- modeldown.py | 1 + src/voxcpm/core.py | 2 +- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/app.py b/app.py index cb4d0fd..1b3241e 100644 --- a/app.py +++ b/app.py @@ -241,7 +241,7 @@ class VoxCPMDemo: if self.voxcpm_model is not None: return self.voxcpm_model logger.info(f"Loading model: {self._model_id}") - self.voxcpm_model = voxcpm.VoxCPM.from_pretrained(self._model_id, optimize=True,local_files_only=True) + self.voxcpm_model = voxcpm.VoxCPM.from_pretrained(self._model_id, optimize=True,zipenhancer_model_id="./models/iic/speech_zipenhancer_ans_multiloss_16k_base") logger.info("Model loaded successfully.") return self.voxcpm_model @@ -488,7 +488,7 @@ def run_demo( server_name: str = "0.0.0.0", server_port: int = 8808, show_error: bool = True, - model_id: str = "./models/openbmb/VoxCPM2", + model_id: str = "./models/OpenBMB/VoxCPM2", ): demo = VoxCPMDemo(model_id=model_id) interface = create_demo_interface(demo) @@ -506,7 +506,7 @@ if __name__ == "__main__": import argparse parser = argparse.ArgumentParser() parser.add_argument( - "--model-id", type=str, default="./models/openbmb/VoxCPM2", + "--model-id", type=str, default="./models/OpenBMB/VoxCPM2", help="本地路径或HuggingFace仓库ID(默认:./models/openbmb/VoxCPM2)", ) parser.add_argument("--port", type=int, default=8808, help="服务端口") diff --git a/modeldown.py b/modeldown.py index d72f8b1..ad485a1 100644 --- a/modeldown.py +++ b/modeldown.py @@ -31,3 +31,4 @@ def download(repo_id:str, local_dir:str): if __name__ == "__main__": download("OpenBMB/VoxCPM2", "./models") download("iic/SenseVoiceSmall", "./models") + download('iic/speech_zipenhancer_ans_multiloss_16k_base',"./models") diff --git a/src/voxcpm/core.py b/src/voxcpm/core.py index 81692c9..365a64d 100644 --- a/src/voxcpm/core.py +++ b/src/voxcpm/core.py @@ -149,7 +149,7 @@ class VoxCPM: if not repo_id: raise ValueError("You must provide hf_model_id") - # Load from local path if provided + # 从本地路径加载(如果提供) if os.path.isdir(repo_id): local_path = repo_id else: