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