InternLM / InternLM/lmdeploy

[Feature] support workers when using lite autoawq

Open
#2,167 1 comment 0 reactions 1 assignee Claimed by @AllentDan View on GitHub
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

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.