Forráskód Böngészése

Update mineru/model/utils/pytorchocr/modeling/heads/rec_ppformulanet_head.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Xiaomeng Zhao 3 hete
szülő
commit
80436a89f9

+ 1 - 1
mineru/model/utils/pytorchocr/modeling/heads/rec_ppformulanet_head.py

@@ -1107,7 +1107,7 @@ class PPFormulaNet_Head(UniMERNetHead):
             model_inputs = self.prepare_inputs_for_generation_export(
                 past_key_values=past_key_values, **model_kwargs
             )
-            decoder_attention_mask = torch.ones(input_ids.shape, device=self.device,)
+            decoder_attention_mask = torch.ones(input_ids.shape, device=self.device)
 
             outputs = self.generate_single_iter(
                 decoder_input_ids=decoder_input_ids,