Merge branch 'alibaba:main' into feat_xqa_1219

Author: qqbbiu

open_source/deps/BUILD

@@ -37,15 +37,6 @@ compile_pip_requirements(
     tags = ["manual"],
 )
 
-compile_pip_requirements(
-    name = "requirements_torch_gpu",
-    src = "requirements_torch_gpu.txt",
-    extra_args = PIP_EXTRA_ARGS,
-    extra_data = ["//open_source/deps:requirements_base.txt"],
-    requirements_txt = "requirements_lock_torch_gpu.txt",
-    tags = ["manual"],
-)
-
 compile_pip_requirements(
     name = "requirements_torch_gpu_cuda12",
     src = "requirements_torch_gpu_cuda12.txt",

open_source/deps/pip.bzl

@@ -25,7 +25,7 @@ def pip_deps():
 
     pip_parse(
         name = "pip_ppu_torch",
-        requirements_lock = "//open_source/deps:requirements_lock_torch_gpu.txt",
+        requirements_lock = "//open_source/deps:requirements_lock_torch_gpu_cuda12.txt",
         python_interpreter = "/opt/conda310/bin/python3",
         extra_pip_args = PIP_EXTRA_ARGS,
         timeout = 3600,

open_source/deps/requirements_base.txt

@@ -53,3 +53,4 @@ typing-extensions
 uvicorn==0.30.0
 xfastertransformer_devel==1.8.1.1 ; platform_machine == "x86_64"
 xfastertransformer_devel_icx==1.8.1.1 ; platform_machine == "x86_64"
+pybind11_stubgen

open_source/deps/requirements_lock_cuda12_arm.txt

@@ -2588,6 +2588,10 @@ pyarrow==21.0.0 \
     --hash=sha256:fc0d2f88b81dcf3ccf9a6ae17f89183762c8a94a5bdcfa09e05cfe413acf0503 \
     --hash=sha256:fee33b0ca46f4c85443d6c450357101e47d53e6c3f008d658c27a2d020d44c79
     # via pyodps
+pybind11-stubgen==2.5.5 \
+    --hash=sha256:10824cd2fc5cbbee032b8fb39e6f6c08de232deb309bc66d786a6c6e8a4601bd \
+    --hash=sha256:758d6d6bbeefc62ad7f78d5e5bbf357ccf6af83cd4504f5f549403f452942708
+    # via -r open_source/deps/requirements_base.txt
 pycparser==2.23 \
     --hash=sha256:78816d4f24add8f10a06d6f05b4d424ad9e96cfebf68a4ddc99c65c0720d00c2 \
     --hash=sha256:e5c6e8d3fbad53479cab09ac03729e0a9faf2bee3db8208a550daf5af81a5934

open_source/deps/requirements_lock_rocm.txt

@@ -2053,6 +2053,10 @@ pybind11==3.0.1 \
     --hash=sha256:9c0f40056a016da59bab516efb523089139fcc6f2ba7e4930854c61efb932051 \
     --hash=sha256:aa8f0aa6e0a94d3b64adfc38f560f33f15e589be2175e103c0a33c6bce55ee89
     # via aiter
+pybind11-stubgen==2.5.5 \
+    --hash=sha256:10824cd2fc5cbbee032b8fb39e6f6c08de232deb309bc66d786a6c6e8a4601bd \
+    --hash=sha256:758d6d6bbeefc62ad7f78d5e5bbf357ccf6af83cd4504f5f549403f452942708
+    # via -r open_source/deps/requirements_base.txt
 pycparser==2.22 \
     --hash=sha256:491c8be9c040f5390f5bf44a5b07752bd07f56edf992381b05c701439eec10f6 \
     --hash=sha256:c3702b6d3dd8c7abc1afa565d7e63d53a1d0bd86cdc24edd75470f4de499cfcc

open_source/deps/requirements_lock_torch_cpu.txt

@@ -6,7 +6,6 @@
 #
 --index-url https://artlab.alibaba-inc.com/1/PYPI/simple/
 --extra-index-url https://mirrors.aliyun.com/pypi/simple/
---trusted-host artlab.alibaba-inc.com
 
 accelerate==0.25.0 \
     --hash=sha256:c7bb817eb974bba0ff3ea1ba0f24d55afb86d50e3d4fe98d6922dc69cf2ccff1 \
@@ -2275,6 +2274,10 @@ pyarrow==17.0.0 \
     # via
     #   datasets
     #   pyodps
+pybind11-stubgen==2.5.5 \
+    --hash=sha256:10824cd2fc5cbbee032b8fb39e6f6c08de232deb309bc66d786a6c6e8a4601bd \
+    --hash=sha256:758d6d6bbeefc62ad7f78d5e5bbf357ccf6af83cd4504f5f549403f452942708
+    # via -r open_source/deps/requirements_base.txt
 pycparser==2.22 \
     --hash=sha256:491c8be9c040f5390f5bf44a5b07752bd07f56edf992381b05c701439eec10f6 \
     --hash=sha256:c3702b6d3dd8c7abc1afa565d7e63d53a1d0bd86cdc24edd75470f4de499cfcc