- 修复路径错误
This commit is contained in:
@@ -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="服务端口")
|
||||||
|
|||||||
@@ -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
@@ -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:
|
||||||
|
|||||||
Reference in New Issue
Block a user