"sgl-kernel/git@developer.sourcefind.cn:change/sglang.git" did not exist on "e0b2d3eebebd3d4efc7e323ad2dee605b607f394"
Commit ae6f6d56 authored by Louis Martin's avatar Louis Martin Committed by Myle Ott
Browse files

Fix call to non-existing to_string method

parent 2f781c5a
......@@ -45,7 +45,7 @@ class Dictionary(object):
Can optionally remove BPE symbols or escape <unk> words.
"""
if torch.is_tensor(tensor) and tensor.dim() == 2:
return '\n'.join(self.to_string(t) for t in tensor)
return '\n'.join(self.string(t) for t in tensor)
def token_string(i):
if i == self.unk():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment