diff --git a/pyproject_gpu.toml b/pyproject_gpu.toml index bd2102dd..c2ed896b 100644 --- a/pyproject_gpu.toml +++ b/pyproject_gpu.toml @@ -91,13 +91,16 @@ detectron2 = [ {markers = "sys_platform != 'win32' and sys_platform != 'linux' ", url = "https://pai-rag.oss-cn-hangzhou.aliyuncs.com/packages/python_wheels/detectron2-0.6%2B864913fpt2.2.2cpu-cp311-cp311-macosx_10_9_universal2.whl"} ] magic-pdf = {version = "0.7.0b1", extras = ["full"]} +paddlepaddle-gpu = [ + {markers = "sys_platform == 'linux'", version = "3.0.0b1", url = "https://paddle-whl.bj.bcebos.com/stable/cu118/paddlepaddle-gpu/paddlepaddle_gpu-3.0.0b1-cp311-cp311-linux_x86_64.whl"}, + {markers = "sys_platform == 'win32'", version = "3.0.0b1", url = "https://paddle-whl.bj.bcebos.com/stable/cu118/paddlepaddle-gpu/paddlepaddle_gpu-3.0.0b1-cp311-cp311-win_amd64.whl"} +] peft = "^0.12.0" aliyun-bootstrap = "^1.0.2" docx = "^0.2.4" python-pptx = "^1.0.2" aspose-slides = "^24.10.0" ultralytics = "8.3.43" -paddlepaddle-gpu = {version = "3.0.0b1", url = "https://www.paddlepaddle.org.cn/packages/stable/cu118/"} [tool.poetry.scripts] pai_rag = "pai_rag.main:run"