Browse Source

fix(setup): update PyMuPDF and paddlepaddle dependencies

myhloli 1 year ago
parent
commit
46d7549926
2 changed files with 13 additions and 3 deletions
  1. 1 1
      requirements.txt
  2. 12 2
      setup.py

+ 1 - 1
requirements.txt

@@ -1,7 +1,7 @@
 boto3>=1.28.43
 Brotli>=1.1.0
 click>=8.1.7
-PyMuPDF>=1.24.7
+PyMuPDF>=1.24.9
 loguru>=0.6.0
 numpy>=1.21.6,<2.0.0
 fast-langdetect==0.2.0

+ 12 - 2
setup.py

@@ -32,8 +32,18 @@ if __name__ == '__main__':
         },
         install_requires=parse_requirements('requirements.txt'),  # 项目依赖的第三方库
         extras_require={
-            "lite": ["paddleocr==2.7.3", "paddlepaddle", "paddlepaddle-gpu"],
-            "full": ["unimernet", "matplotlib", "ultralytics", "paddleocr==2.7.3", "paddlepaddle", "paddlepaddle-gpu"],
+            "lite": ["paddleocr==2.7.3",
+                     "paddlepaddle==3.0.0b1;platform_system=='Linux'",
+                     "paddlepaddle==2.6.1;platform_system=='Windows' or platform_system=='Darwin'",
+                     ],
+            "full": ["unimernet",
+                     "matplotlib",
+                     "ultralytics",
+                     "paddleocr==2.7.3",
+                     "paddlepaddle==3.0.0b1;platform_system=='Linux'",
+                     "paddlepaddle==2.6.1;platform_system=='Windows' or platform_system=='Darwin'",
+                     "detectron2"
+                     ],
         },
         description="A practical tool for converting PDF to Markdown",  # 简短描述
         long_description=long_description,  # 详细描述