فهرست منبع

refactor: simplify UnimernetModel constructor by removing unused cfg_path parameter

myhloli 5 ماه پیش
والد
کامیت
9ded9e6bad
1فایلهای تغییر یافته به همراه1 افزوده شده و 1 حذف شده
  1. 1 1
      mineru/model/mfr/unimernet/Unimernet.py

+ 1 - 1
mineru/model/mfr/unimernet/Unimernet.py

@@ -19,7 +19,7 @@ class MathDataset(Dataset):
 
 
 class UnimernetModel(object):
-    def __init__(self, weight_dir, cfg_path, _device_="cpu"):
+    def __init__(self, weight_dir, _device_="cpu"):
         from .unimernet_hf import UnimernetModel
         if _device_.startswith("mps"):
             self.model = UnimernetModel.from_pretrained(weight_dir, attn_implementation="eager")