[Feature] support workers when using lite autoawq
Open
- Dominant language
- Python
- Stars
- 8.1k
- Forks
- 748
- Avg merge
- 6d 2h
- Merged PRs (30d)
- 54
Description
### Motivation
Hi all. @lvhan028 @lzhangzz @AllentDan When I want to use `lite autoawq` with Llama 3.1 405B Instruct, it takes a very long time.
```bash
python3 -m lmdeploy lite auto_awq Meta-Llama-3.1-405B-Instruct
```
May we support `--workers` similar to TensorRT LLM? Thanks.
ref https://nvidia.github.io/TensorRT-LLM/performance/perf-overview.html#engine-building
> --workers affects the number of threads that build the engine file and does not necessarily need to match the TP size.
### Related resources
_No response_
### Additional context
_No response_
Contributor guide
Assessment
This issue has not been assessed yet.