浏览代码

build(deps): update dependencies and add support for old Linux systems

- Update transformers to exclude version 4.51.0 due to compatibility issues- Rapid table version range expanded to >=1.0.5,<2.0.0
- Add separate 'full_old_linux' extras_require for better support of older Linux systems
- Update matplotlib version requirements for different platforms
- Remove platform-specific paddlepaddle versions,
myhloli 7 月之前
父节点
当前提交
9b5d2796f8
共有 2 个文件被更改,包括 22 次插入10 次删除
  1. 1 1
      requirements.txt
  2. 21 9
      setup.py

+ 1 - 1
requirements.txt

@@ -9,7 +9,7 @@ PyMuPDF>=1.24.9,<1.25.0
 scikit-learn>=1.0.2
 torch>=2.2.2,!=2.5.0,!=2.5.1,<=2.6.0
 torchvision
-transformers>=4.49.0,<5.0.0
+transformers>=4.49.0,!=4.51.0,<5.0.0
 pdfminer.six==20231228
 tqdm>=4.67.1
 # The requirements.txt must ensure that only necessary external dependencies are introduced. If there are new dependencies to add, please contact the project administrator.

+ 21 - 9
setup.py

@@ -33,27 +33,39 @@ if __name__ == '__main__':
         },
         install_requires=parse_requirements('requirements.txt'),  # 项目依赖的第三方库
         extras_require={
-            "lite": ["paddleocr==2.7.3",
-                     "paddlepaddle==3.0.0b1;platform_system=='Linux'",
-                     "paddlepaddle==2.6.1;platform_system=='Windows' or platform_system=='Darwin'",
-                     ],
+            "lite": [
+                    "paddleocr==2.7.3",
+                    "paddlepaddle==3.0.0b1;platform_system=='Linux'",
+                    "paddlepaddle==2.6.1;platform_system=='Windows' or platform_system=='Darwin'",
+            ],
             "full": [
                      "matplotlib<=3.9.0;platform_system=='Windows'",  # 3.9.1及之后不提供windows的预编译包,避免一些没有编译环境的windows设备安装失败
                      "matplotlib>=3.10;platform_system=='Linux' or platform_system=='Darwin'",  # linux 和 macos 不应限制matplotlib的最高版本,以避免无法更新导致的一些bug
                      "ultralytics>=8.3.48",  # yolov8,公式检测
                      "doclayout_yolo==0.0.2b1",  # doclayout_yolo
                      "dill>=0.3.9,<1",  # doclayout_yolo
-                     "rapid_table>=1.0.3,<2.0.0",  # rapid_table
+                     "rapid_table>=1.0.5,<2.0.0",  # rapid_table
                      "PyYAML>=6.0.2,<7",  # yaml
                      "ftfy>=6.3.1,<7", # unimernet_hf
                      "openai>=1.70.0,<2",  # openai SDK
                      "shapely>=2.0.7,<3",  # imgaug-paddleocr2pytorch
                      "pyclipper>=1.3.0,<2",  # paddleocr2pytorch
                      "omegaconf>=2.3.0,<3",  # paddleocr2pytorch
-                     ],
-            "old_linux":[
-                "albumentations<=1.4.20", # 1.4.21引入的simsimd不支持2019年及更早的linux系统
-                "rapid_table==1.0.3",  # rapid_table新版本依赖的onnxruntime不支持2019年及更早的linux系统
+            ],
+            "full_old_linux":[
+                    "matplotlib>=3.10",
+                    "ultralytics>=8.3.48",  # yolov8,公式检测
+                    "doclayout_yolo==0.0.2b1",  # doclayout_yolo
+                    "dill>=0.3.9,<1",  # doclayout_yolo
+                    "rapid_table>=1.0.3,<2.0.0",  # rapid_table
+                    "PyYAML>=6.0.2,<7",  # yaml
+                    "ftfy>=6.3.1,<7",  # unimernet_hf
+                    "openai>=1.70.0,<2",  # openai SDK
+                    "shapely>=2.0.7,<3",  # imgaug-paddleocr2pytorch
+                    "pyclipper>=1.3.0,<2",  # paddleocr2pytorch
+                    "omegaconf>=2.3.0,<3",  # paddleocr2pytorch
+                    "albumentations<=1.4.20", # 1.4.21引入的simsimd不支持2019年及更早的linux系统
+                    "rapid_table==1.0.3",  # rapid_table新版本依赖的onnxruntime不支持2019年及更早的linux系统
             ],
         },
         description="A practical tool for converting PDF to Markdown",  # 简短描述