Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "2a3fbc2cc269aa0c0d5cfdfaa3564d769d92b882"
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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
fairseq/dictionary.py
View file @ ae6f6d56
...@@ -45,7 +45,7 @@ class Dictionary(object): ...@@ -45,7 +45,7 @@ class Dictionary(object):
Can optionally remove BPE symbols or escape <unk> words. Can optionally remove BPE symbols or escape <unk> words.
""" """
if torch.is_tensor(tensor) and tensor.dim() == 2: 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): def token_string(i):
if i == self.unk(): 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