adapt to npu

warrentdrew · warrentdrew · commit f897ca408f01 · 2025-03-04T18:27:40.000+08:00
diff --git a/ppocr/modeling/heads/rec_latexocr_head.py b/ppocr/modeling/heads/rec_latexocr_head.py
@@ -786,7 +786,7 @@ def forward(
             x, mask=mask, mems=mems, return_hiddens=True, seq_len=seq_len, **kwargs
         )
         x = self.norm(x)
-        mem, x = x[:, :num_mem], x[:, num_mem:]
+        x = x[:, num_mem:]
         out = self.to_logits(x) if not return_embeddings else x
         if return_mems:
             hiddens = intermediates.hiddens

Original file line number	Diff line number	Diff line change
`@@ -786,7 +786,7 @@ def forward(`
`786`	`786`	`x, mask=mask, mems=mems, return_hiddens=True, seq_len=seq_len, **kwargs`
`787`	`787`	`)`
`788`	`788`	`x = self.norm(x)`
`789`		`- mem, x = x[:, :num_mem], x[:, num_mem:]`
	`789`	`+ x = x[:, num_mem:]`
`790`	`790`	`out = self.to_logits(x) if not return_embeddings else x`
`791`	`791`	`if return_mems:`
`792`	`792`	`hiddens = intermediates.hiddens`