"examples/custom_diffusion/train_custom_diffusion.py" did not exist on "522f8aa7b2599e58b9815c7d81e9136797c68b5c"
Commit 0f4d92b6 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

fix conflict wiith SRN

parent 7a054c85
......@@ -255,12 +255,12 @@ class SRNLabelDecode(BaseRecLabelDecode):
text = self.decode(preds_idx, preds_prob)
if label is None:
text = self.decode(preds_idx, preds_prob, is_remove_duplicate=True)
text = self.decode(preds_idx, preds_prob, is_remove_duplicate=False)
return text
label = self.decode(label)
return text, label
def decode(self, text_index, text_prob=None, is_remove_duplicate=True):
def decode(self, text_index, text_prob=None, is_remove_duplicate=False):
""" convert text-index into text-label. """
result_list = []
ignored_tokens = self.get_ignored_tokens()
......
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