reset latex ocr (PaddlePaddle#14046)

Liyulingyue · Oct 18, 2024 · de45732 · de45732
1 parent de82b7b
commit de45732
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ppocr/modeling/heads/rec_latexocr_head.py b/ppocr/modeling/heads/rec_latexocr_head.py
@@ -948,15 +948,15 @@ def generate_export(
         b, t = start_tokens.shape
 
         self.net.eval()
-        out_tmp = start_tokens
+        out = start_tokens
         mask = kwargs.pop("mask", None)
 
         if mask is None:
-            mask = paddle.full_like(out_tmp, True, dtype=paddle.bool)
+            mask = paddle.full_like(out, True, dtype=paddle.bool)
 
         i_idx = paddle.full([], 0)
         while i_idx < paddle.to_tensor(seq_len):
-            x = out_tmp[:, -self.max_seq_len :]
+            x = out[:, -self.max_seq_len :]
             paddle.jit.api.set_dynamic_shape(x, [-1, -1])
             mask = mask[:, -self.max_seq_len :]
             paddle.jit.api.set_dynamic_shape(mask, [-1, -1])
@@ -969,7 +969,7 @@ def generate_export(
                 probs = F.softmax(filtered_logits / temperature, axis=-1)
 
             sample = paddle.multinomial(probs, 1)
-            out = paddle.concat((out_tmp, sample), axis=-1)
+            out = paddle.concat((out, sample), axis=-1)
 
             pad_mask = paddle.full(shape=[mask.shape[0], 1], fill_value=1, dtype="bool")
             mask = paddle.concat((mask, pad_mask), axis=1)