[Feature] Support `return_hidden_states` in async engine

Métricas do repositório

Stars

28.442 estrelas (28.442 estrelas)

Métricas de merge de PR

Mesclagem média 1d 13h (Mesclagem média 1d 13h)1.000 fundiu PRs em 30d (1.000 fundiu PRs em 30d)

Description

result = await model.async_generate(
        ["hello", "how are you doing?"],
        sampling_params={"temperature": 0, "max_new_tokens": 1},
        return_hidden_states=True)

TypeError: Engine.async_generate() got an unexpected keyword argument 'return_hidden_states'

Guia do colaborador

Direção de pesquisa

Examine como return hidden states é tratado em Engine.generate e adicione o mesmo suporte a async generate.

Pilha de tecnologia

python

Domain

backend

Tipo Issue

Funcionalidade

DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.

2

Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.

1-3 horas

Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.

Ativo

ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.

Claro

Prerequisites

Python

Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.

70