Fix bug in symm mem pre-allocation default (#19082)
This commit is contained in:
@@ -1069,10 +1069,7 @@ class ServerArgs:
|
|||||||
self.adjust_mem_fraction_for_vlm(model_config)
|
self.adjust_mem_fraction_for_vlm(model_config)
|
||||||
|
|
||||||
# If symm mem is enabled and prealloc size is not set, set it to 4GB
|
# If symm mem is enabled and prealloc size is not set, set it to 4GB
|
||||||
if (
|
if self.enable_symm_mem and not envs.SGLANG_SYMM_MEM_PREALLOC_GB_SIZE.is_set():
|
||||||
self.enable_symm_mem
|
|
||||||
and not envs.SGLANG_SYMM_MEM_PREALLOC_GB_SIZE.is_set()
|
|
||||||
):
|
|
||||||
envs.SGLANG_SYMM_MEM_PREALLOC_GB_SIZE.set(4)
|
envs.SGLANG_SYMM_MEM_PREALLOC_GB_SIZE.set(4)
|
||||||
logger.warning(
|
logger.warning(
|
||||||
"Symmetric memory is enabled, setting symmetric memory prealloc size to 4GB as default."
|
"Symmetric memory is enabled, setting symmetric memory prealloc size to 4GB as default."
|
||||||
|
|||||||
Reference in New Issue
Block a user