Docker 启动示例
Docker 启动示例
docker run --gpus all --shm-size 1g -p 8080:80 -v $PWD/data:/data ghcr.io/huggingface/text-generation-inference:latest --model-id <模型名或路径> --max-concurrent-requests 64| 端口 | 容器内 80,示例映射为 8080 |
|---|---|
| 并发限制 | MAX_CONCURRENT_REQUESTS / --max-concurrent-requests |
| 批处理 | --max-batch-total-tokens 控制批内 token 预算 |
--shm-size 过小会在较大批量推理时报共享内存错误,示例给了 1g。