huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 12:24:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix seq2seq.

Browse Source
This commit is contained in:
Nicolas Patry 2024-01-26 17:34:38 +00:00
parent 0452d590d0
commit a5600c23af
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/text_generation_server/models/seq2seq_lm.py
View File

@ -641,7 +641,7 @@ class Seq2SeqLM(Model):
) )
# Speculation is not active for seq2seq # Speculation is not active for seq2seq
accepted_ids = torch.ones_like(batch.input_ids) accepted_ids = torch.ones_like(batch.decoder_input_ids)
batch_top_token_ids, batch_top_token_logprobs = batch_top_tokens( batch_top_token_ids, batch_top_token_logprobs = batch_top_tokens(
batch.top_n_tokens, batch.top_n_tokens,
batch.top_n_tokens_tensor, batch.top_n_tokens_tensor,