debug

clovaai · May 15, 2024 · c8d3618 · c8d3618
1 parent 5c652e1
commit c8d3618
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/donut/model.py b/donut/model.py
@@ -799,7 +799,7 @@ def token2json_with_confs(
                 ]
                 tokens = tokens.replace(start_token, "")
                 tksplit = [tk for tk in tokens.split(delim) if tk]
-                print(len(tksplit), len(confs), len(idxs))
+                print(tokens, tksplit, idxs, len(tksplit), len(confs), len(idxs))
                 assert len(tksplit) == len(confs) == len(idxs)
             else:
                 end_token = end_token.group()