update poetry

aigc-apps · Dec 6, 2024 · a7bead7 · a7bead7
1 parent fc29032
commit a7bead7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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"