EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Tune params (#696)

Browse Source
This commit is contained in:
Ying Sheng
2024-07-22 03:19:24 -07:00
committed by GitHub
parent d198791fe8
commit c3f1aac811
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
python/sglang/srt/server_args.py
View File

@@ -70,15 +70,15 @@ class ServerArgs:
self.tokenizer_path = self.model_path self.tokenizer_path = self.model_path
if self.mem_fraction_static is None: if self.mem_fraction_static is None:
if self.tp_size >= 16: if self.tp_size >= 16:
self.mem_fraction_static = 0.74 self.mem_fraction_static = 0.80
elif self.tp_size >= 8: elif self.tp_size >= 8:
self.mem_fraction_static = 0.78 self.mem_fraction_static = 0.84
elif self.tp_size >= 4: elif self.tp_size >= 4:
self.mem_fraction_static = 0.82 self.mem_fraction_static = 0.86
elif self.tp_size >= 2: elif self.tp_size >= 2:
self.mem_fraction_static = 0.85
else:
self.mem_fraction_static = 0.88 self.mem_fraction_static = 0.88
else:
self.mem_fraction_static = 0.89
if isinstance(self.additional_ports, int): if isinstance(self.additional_ports, int):
self.additional_ports = [self.additional_ports] self.additional_ports = [self.additional_ports]
elif self.additional_ports is None: elif self.additional_ports is None: