Explorar o código

fix post process in unimernet (#4213)

liuhongen1234567 hai 5 meses
pai
achega
718c3a9c6c
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      paddlex/inference/models/formula_recognition/processors.py

+ 1 - 1
paddlex/inference/models/formula_recognition/processors.py

@@ -798,7 +798,7 @@ class UniMERNetDecode(object):
             for i in reversed(range(len(toks[b]))):
                 if toks[b][i] is None:
                     toks[b][i] = ""
-                toks[b][i] = toks[b][i].replace("", " ").strip()
+                toks[b][i] = toks[b][i].replace("Ġ", " ").strip()
                 if toks[b][i] in (
                     [
                         self.tokenizer.bos_token,