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: