[HiCache] Add an env var to control transfer engine reuse (#19867)
This commit is contained in:
@@ -289,6 +289,7 @@ class Envs:
|
|||||||
|
|
||||||
# Mooncake Store
|
# Mooncake Store
|
||||||
SGLANG_HICACHE_MOONCAKE_CONFIG_PATH = EnvStr(None)
|
SGLANG_HICACHE_MOONCAKE_CONFIG_PATH = EnvStr(None)
|
||||||
|
SGLANG_HICACHE_MOONCAKE_REUSE_TE = EnvBool(True)
|
||||||
MOONCAKE_MASTER = EnvStr(None)
|
MOONCAKE_MASTER = EnvStr(None)
|
||||||
MOONCAKE_CLIENT = EnvStr(None)
|
MOONCAKE_CLIENT = EnvStr(None)
|
||||||
MOONCAKE_LOCAL_HOSTNAME = EnvStr("localhost")
|
MOONCAKE_LOCAL_HOSTNAME = EnvStr("localhost")
|
||||||
|
|||||||
@@ -868,6 +868,7 @@ class ModelRunner(ModelRunnerKVCacheMixin):
|
|||||||
or (
|
or (
|
||||||
self.server_args.enable_hierarchical_cache
|
self.server_args.enable_hierarchical_cache
|
||||||
and self.server_args.hicache_storage_backend == "mooncake"
|
and self.server_args.hicache_storage_backend == "mooncake"
|
||||||
|
and envs.SGLANG_HICACHE_MOONCAKE_REUSE_TE.get()
|
||||||
)
|
)
|
||||||
or (
|
or (
|
||||||
self.server_args.encoder_only
|
self.server_args.encoder_only
|
||||||
|
|||||||
Reference in New Issue
Block a user