Organize server_args (#277)
This commit is contained in:
@@ -46,7 +46,6 @@ class ModelRpcServer(rpyc.Service):
|
||||
server_args, port_args = [obtain(x) for x in [server_args, port_args]]
|
||||
|
||||
# Copy arguments
|
||||
self.model_mode = server_args.model_mode
|
||||
self.tp_rank = tp_rank
|
||||
self.tp_size = server_args.tp_size
|
||||
self.schedule_heuristic = server_args.schedule_heuristic
|
||||
@@ -61,15 +60,22 @@ class ModelRpcServer(rpyc.Service):
|
||||
server_args.trust_remote_code,
|
||||
context_length=server_args.context_length,
|
||||
)
|
||||
|
||||
# for model end global settings
|
||||
server_args_dict = {
|
||||
"enable_flashinfer": server_args.enable_flashinfer,
|
||||
"attention_reduce_in_fp32": server_args.attention_reduce_in_fp32,
|
||||
}
|
||||
|
||||
self.model_runner = ModelRunner(
|
||||
model_config=self.model_config,
|
||||
mem_fraction_static=server_args.mem_fraction_static,
|
||||
tp_rank=tp_rank,
|
||||
tp_size=server_args.tp_size,
|
||||
nccl_port=port_args.nccl_port,
|
||||
server_args=server_args,
|
||||
load_format=server_args.load_format,
|
||||
trust_remote_code=server_args.trust_remote_code,
|
||||
server_args_dict=server_args_dict,
|
||||
)
|
||||
if is_multimodal_model(server_args.model_path):
|
||||
self.processor = get_processor(
|
||||
@@ -104,11 +110,11 @@ class ModelRpcServer(rpyc.Service):
|
||||
f"max_total_num_token={self.max_total_num_token}, "
|
||||
f"max_prefill_num_token={self.max_prefill_num_token}, "
|
||||
f"context_len={self.model_config.context_len}, "
|
||||
f"model_mode={self.model_mode}"
|
||||
)
|
||||
logger.info(server_args.get_optional_modes_logging())
|
||||
|
||||
# Init cache
|
||||
self.tree_cache = RadixCache(disable="no-cache" in self.model_mode)
|
||||
self.tree_cache = RadixCache(server_args.disable_radix_cache)
|
||||
self.tree_cache_metrics = {"total": 0, "hit": 0}
|
||||
self.scheduler = Scheduler(
|
||||
self.schedule_heuristic,
|
||||
|
||||
Reference in New Issue
Block a user