Skip to content

Commit c008a58

Browse files
authored
fix
1 parent 7ca7b0d commit c008a58

File tree

1 file changed

+9
-5
lines changed

1 file changed

+9
-5
lines changed

lightllm/server/router/model_infer/mode_backend/base_backend.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -237,11 +237,15 @@ def init_model(self, kvargs):
237237
raise e
238238

239239
set_random_seed(2147483647)
240-
self.radix_cache = RadixCache(
241-
get_unique_server_name(),
242-
self.model.mem_manager.size,
243-
self.rank_in_node,
244-
mem_manager=self.model.mem_manager,
240+
self.radix_cache = (
241+
RadixCache(
242+
get_unique_server_name(),
243+
self.model.mem_manager.size,
244+
self.rank_in_node,
245+
mem_manager=self.model.mem_manager,
246+
)
247+
if self.use_dynamic_prompt_cache
248+
else None
245249
)
246250

247251
if "prompt_cache_kv_buffer" in model_cfg:

0 commit comments

Comments
 (0)