Add source for onnxruntime-gpu cuda 12 support

Signed-off-by: declark1 <[email protected]>

Author: declark1

server/pyproject.toml

@@ -24,7 +24,7 @@ texttable = { version = "^1.7.0", optional = true }
 transformers = "4.38.2"
 optimum = { version = "^1.17.1", extras = ["onnxruntime-gpu"], optional = true }
 onnxruntime = { version = "^1.17.1", optional = true }
-onnxruntime-gpu = { version = "^1.17.1", optional = true }
+onnxruntime-gpu = { version = "^1.17.1", source = "onnxruntime-gpu-src", optional = true }
 onnx = { version = "^1.15.0", optional = true }
 einops = "^0.7.0"
 ibm-fms = { version = "^0.0", optional = true }
@@ -51,6 +51,12 @@ quantize = ["datasets", "texttable"]
 grpcio-tools = "^1.62.1"
 pytest = "^8.0.1"
 
+
+[[tool.poetry.source]]
+name = "onnxruntime-gpu-src"
+url = "https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/"
+priority = "explicit"
+
 [build-system]
 requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"