- 修复路径错误

This commit is contained in:
Pine
2026-05-06 14:58:46 +08:00
parent 525ebc65a3
commit 2e4c601d1d
3 changed files with 5 additions and 4 deletions
+3 -3
View File
@@ -241,7 +241,7 @@ class VoxCPMDemo:
if self.voxcpm_model is not None: if self.voxcpm_model is not None:
return self.voxcpm_model return self.voxcpm_model
logger.info(f"Loading model: {self._model_id}") 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.") logger.info("Model loaded successfully.")
return self.voxcpm_model return self.voxcpm_model
@@ -488,7 +488,7 @@ def run_demo(
server_name: str = "0.0.0.0", server_name: str = "0.0.0.0",
server_port: int = 8808, server_port: int = 8808,
show_error: bool = True, show_error: bool = True,
model_id: str = "./models/openbmb/VoxCPM2", model_id: str = "./models/OpenBMB/VoxCPM2",
): ):
demo = VoxCPMDemo(model_id=model_id) demo = VoxCPMDemo(model_id=model_id)
interface = create_demo_interface(demo) interface = create_demo_interface(demo)
@@ -506,7 +506,7 @@ if __name__ == "__main__":
import argparse import argparse
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument( 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", help="本地路径或HuggingFace仓库ID(默认:./models/openbmb/VoxCPM2",
) )
parser.add_argument("--port", type=int, default=8808, help="服务端口") parser.add_argument("--port", type=int, default=8808, help="服务端口")
+1
View File
@@ -31,3 +31,4 @@ def download(repo_id:str, local_dir:str):
if __name__ == "__main__": if __name__ == "__main__":
download("OpenBMB/VoxCPM2", "./models") download("OpenBMB/VoxCPM2", "./models")
download("iic/SenseVoiceSmall", "./models") download("iic/SenseVoiceSmall", "./models")
download('iic/speech_zipenhancer_ans_multiloss_16k_base',"./models")
+1 -1
View File
@@ -149,7 +149,7 @@ class VoxCPM:
if not repo_id: if not repo_id:
raise ValueError("You must provide hf_model_id") raise ValueError("You must provide hf_model_id")
# Load from local path if provided # 从本地路径加载(如果提供)
if os.path.isdir(repo_id): if os.path.isdir(repo_id):
local_path = repo_id local_path = repo_id
else: else: