InternLM / InternLM/lmdeploy

[Bug] 不支持qwen0.5b的加速?以及qwen0.5b的awq量化?

Open
#1,870 5 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
8.1k
Forks
748
Avg merge
6d 2h
Merged PRs (30d)
54

Description

### Checklist

- [ ] 1. I have searched related issues but cannot get the expected help.
- [ ] 2. The bug has not been fixed in the latest version.

### Describe the bug

是否不支持qwen0.5b的加速?以及qwen0.5b的awq量化?
qwen0.5b 单卡t4 推理延时如下:
vllm:1.3s
lmdeploy:3.2s

### Reproduction

from lmdeploy import pipeline, GenerationConfig, TurbomindEngineConfig
backend_config = TurbomindEngineConfig(cache_max_entry_count=0.9)
gen_config = GenerationConfig(top_p=0.95,
temperature=0.001,
max_new_tokens=512)

llm = pipeline(model_path,
backend_config=backend_config)

arrival_time = time.time()
llm_results = llm(prompts, gen_config=gen_config)
finished_time = time.time()

### Environment

```Shell
lmdeploy 0.4.2
```

### Error traceback

_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.