Skip to content

Commit 0734973

Browse files
fix: check for tokenizer eos_token in ModelInfo response (IBM#93)
Use consistent logic to determine the eos_token_id in ModelInfo as it is in other functions by falling back to the tokenizer's `eos_token_id` attribute if the model config does not have an `eos_token_id`. Fixes the behavior for a model that does not have an eos_token_id in the model config Resolves IBM#91 Signed-off-by: Travis Johnson <[email protected]>
1 parent 2358566 commit 0734973

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

server/text_generation_server/server.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ async def ModelInfo(self, request: generate_pb2.ModelInfoRequest, context) -> ge
7979
return generate_pb2.ModelInfoResponse(
8080
model_type=ModelInfoResponse.ModelType.SEQ2SEQ_LM
8181
if isinstance(self.model, Seq2SeqLM) else ModelInfoResponse.ModelType.CAUSAL_LM,
82-
eos_token=self.model.config.eos_token_id,
82+
eos_token=getattr(self.model.tokenizer, 'model_eos_token_id', self.model.tokenizer.eos_token_id),
8383
batch_padding=not isinstance(self.model, FlashCausalLM),
8484
memory_scaling_model=self.memory_scaling_model,
8585
)

0 commit comments

Comments
 (0)