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

fix: linter

Browse Source
This commit is contained in:
drbh 2025-05-06 00:03:17 +00:00
parent 783ca66926
commit 551ee3a365
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/vlm_causal_lm.py
View File

@ -418,7 +418,7 @@ class VlmCausalLM(FlashCausalLM):
logger.warning( logger.warning(
f"Tokenizer vocab size {self.tokenizer.vocab_size} does not match model vocab size {self.config.vocab_size}. Updating tokenizer vocab size." f"Tokenizer vocab size {self.tokenizer.vocab_size} does not match model vocab size {self.config.vocab_size}. Updating tokenizer vocab size."
) )
# TODO: HUGE HACK! This is a workaround to update the vocab size # TODO: HUGE HACK! This is a workaround to update the vocab size
# in the tokenizer. When the tokenizer is updated within the model # in the tokenizer. When the tokenizer is updated within the model
# the vocab size is not updated in the tokenizer. # the vocab size is not updated in the tokenizer.
self.tokenizer._vocab_size = self.config.vocab_size self.tokenizer._vocab_size = self.config.vocab_size