Spaces:
Runtime error
Runtime error
Update model_server.py
Browse files- model_server.py +2 -1
model_server.py
CHANGED
|
@@ -140,5 +140,6 @@ if __name__ == "__main__":
|
|
| 140 |
parser.add_argument("--model-path", type=str, default="THUDM/glm-4-voice-9b")
|
| 141 |
args = parser.parse_args()
|
| 142 |
|
| 143 |
-
worker = ModelWorker(args.model_path
|
|
|
|
| 144 |
uvicorn.run(app, host=args.host, port=args.port, log_level="info")
|
|
|
|
| 140 |
parser.add_argument("--model-path", type=str, default="THUDM/glm-4-voice-9b")
|
| 141 |
args = parser.parse_args()
|
| 142 |
|
| 143 |
+
worker = ModelWorker(args.model_path)
|
| 144 |
+
#worker = ModelWorker(args.model_path, args.dtype, args.device)
|
| 145 |
uvicorn.run(app, host=args.host, port=args.port, log_level="info")
|