From 1aeebcc7e0c8d6dbded1a6945ea4e4312265eb31 Mon Sep 17 00:00:00 2001 From: wangzaijun Date: Tue, 1 Jul 2025 14:47:22 +0800 Subject: [PATCH] fix --- lightllm/server/embed_cache/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lightllm/server/embed_cache/utils.py b/lightllm/server/embed_cache/utils.py index f2c91a024..6df031293 100644 --- a/lightllm/server/embed_cache/utils.py +++ b/lightllm/server/embed_cache/utils.py @@ -22,7 +22,7 @@ def tensor2bytes(t: torch.Tensor): def bytes2tensor(b): # return torch.from_numpy(np.frombuffer(b, dtype=np.float16)).cuda() - return torch.load(BytesIO(b)) + return torch.load(BytesIO(b), weights_only=False) def create_shm(name, data):