浏览代码

refactor: remove commented-out device assignment in predict_formula.py

myhloli 4 周之前
父节点
当前提交
f57a6d8d9e
共有 1 个文件被更改,包括 0 次插入1 次删除
  1. 0 1
      mineru/model/mfr/pp_formulanet_plus_m/predict_formula.py

+ 0 - 1
mineru/model/mfr/pp_formulanet_plus_m/predict_formula.py

@@ -45,7 +45,6 @@ class FormulaRecognizer(BaseOCRV20):
         super(FormulaRecognizer, self).__init__(network_config)
 
         self.load_state_dict(weights)
-        # device = "cpu"
         self.device = torch.device(device) if isinstance(device, str) else device
         self.net.to(self.device)
         self.net.eval()