[CI] Append test variant name to markdown report header in nightly test (#16166)
This commit is contained in:
@@ -1767,11 +1767,13 @@ class ModelLaunchSettings:
|
||||
tp_size: int = 1,
|
||||
extra_args: Optional[List[str]] = None,
|
||||
env: Optional[dict] = None,
|
||||
variant: Optional[str] = None,
|
||||
):
|
||||
self.model_path = model_path
|
||||
self.tp_size = tp_size
|
||||
self.extra_args = list(extra_args) if extra_args else []
|
||||
self.env = env
|
||||
self.variant = variant
|
||||
|
||||
if self.tp_size > 1 and "--tp" not in self.extra_args:
|
||||
self.extra_args.extend(["--tp", str(self.tp_size)])
|
||||
|
||||
Reference in New Issue
Block a user