InternLM / InternLM/lmdeploy

[Feature] Skip useless computation

Open
#3,663 2 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

### Motivation

I found that the logit in lmdeploy was directly from the last lm_head. Can we skip useless computation from the last attn to lm_head for unwanted logits? llama.cpp applied this for example https://github.com/ggml-org/llama.cpp/blob/5d46babdc2d4675d96ebcf23cac098a02f0d30cc/src/llama-model.cpp#L6888
@lvhan028

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