mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-06 19:40:28 +00:00
[Inference] Optimize request handler of llama (#5512)
* optimize request_handler * fix ways of writing
This commit is contained in:
@@ -298,8 +298,8 @@ class RequestHandler:
|
||||
"""
|
||||
# do logit processor
|
||||
# NOTE: need to decide the granularity to process logits (sequence or batch)
|
||||
config_dict = generation_config.to_dict()
|
||||
for type in ["top_k", "top_p", "min_p"]:
|
||||
config_dict = generation_config.to_dict()
|
||||
if type in config_dict and config_dict[type] is not None:
|
||||
logits = logit_processor(type, logits, config_dict[type])
|
||||
|
||||
|
Reference in New Issue
Block a user