[BugFix] add verify logit_bias to avoid crash because of IndexError (#7749)
This commit is contained in:
@@ -604,7 +604,7 @@ class TokenizerManager:
|
||||
sampling_kwargs = obj.sampling_params
|
||||
sampling_params = SamplingParams(**sampling_kwargs)
|
||||
sampling_params.normalize(self.tokenizer)
|
||||
sampling_params.verify()
|
||||
sampling_params.verify(self.model_config.vocab_size)
|
||||
|
||||
# Build return object
|
||||
if isinstance(obj, GenerateReqInput):
|
||||
|
||||
Reference in New Issue
Block a user