[mypy] Enable type checking for more directories (#29674)
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
@@ -1954,7 +1954,9 @@ class EngineArgs:
|
||||
self.enable_prefix_caching = False
|
||||
|
||||
def _set_default_max_num_seqs_and_batched_tokens_args(
|
||||
self, usage_context: UsageContext, model_config: ModelConfig
|
||||
self,
|
||||
usage_context: UsageContext | None,
|
||||
model_config: ModelConfig,
|
||||
):
|
||||
world_size = self.pipeline_parallel_size * self.tensor_parallel_size
|
||||
(
|
||||
|
||||
Reference in New Issue
Block a user