feat: auto-select model based on device (GPU=fun-asr-nano, CPU=sensevoice)

Author: LauraGPT

funasr/bin/_server_app.py

@@ -37,7 +37,9 @@
 }
 
 
-def create_app(device: str = "cuda", preload_model: str = "sensevoice") -> FastAPI:
+def create_app(device: str = "cuda", preload_model: str = "auto") -> FastAPI:
+    if preload_model == "auto":
+        preload_model = "fun-asr-nano" if device.startswith("cuda") else "sensevoice"
     app = FastAPI(title="FunASR Server", version="1.3.2")
     app.state.models = {}
     app.state.device = device

funasr/bin/server.py

@@ -31,7 +31,7 @@ def main():
     parser.add_argument("--host", default="0.0.0.0", help="Bind address (default: 0.0.0.0)")
     parser.add_argument("--port", type=int, default=8000, help="Port (default: 8000)")
     parser.add_argument("--device", default="cuda", help="Device: cuda, cpu, mps (default: cuda)")
-    parser.add_argument("--model", default="sensevoice", help="Pre-load model: sensevoice, paraformer, fun-asr-nano (default: sensevoice)")
+    parser.add_argument("--model", default="auto", help="Pre-load model: auto (GPU=fun-asr-nano, CPU=sensevoice), sensevoice, paraformer, fun-asr-nano")
     args = parser.parse_args()
 
     try: