We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f7dfc16 commit f9a0111Copy full SHA for f9a0111
lightllm/server/embed_cache/utils.py
@@ -22,7 +22,7 @@ def tensor2bytes(t: torch.Tensor):
22
23
def bytes2tensor(b):
24
# return torch.from_numpy(np.frombuffer(b, dtype=np.float16)).cuda()
25
- return torch.load(BytesIO(b))
+ return torch.load(BytesIO(b), weights_only=False)
26
27
28
def create_shm(name, data):
0 commit comments