InternLM / InternLM/lmdeploy

单条样本推理可以不使用stream_infer吗

Open
#1,891 4 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

- [X] 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

案例代码都是使用stream_infer推理,但是单条样本推理我发现有decode的代码,输出都是id,请问单条样本的预测代码是什么呢

### Reproduction

logits = self.generator.decode(input_ids)

### Environment

```Shell
环境没问题
```

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