Pinning of dependencies to support cu128 with arm64

Signed-off-by: Viktor Kuropiatnyk <vku@zurich.ibm.com>
Removed pr image pushing as it requires a different setup
2025-11-29 08:33:50 +00:00 · 2025-10-08 14:31:40 +02:00 · 2025-10-08 09:59:31 +02:00 · 2025-10-07 16:21:44 +02:00
4 changed files with 1254 additions and 482 deletions
--- a/.github/workflows/ci-images-dryrun.yml
+++ b/.github/workflows/ci-images-dryrun.yml
@@ -32,7 +32,7 @@ jobs:
          - name: docling-project/docling-serve-cu128
            build_args: |
              UV_SYNC_EXTRA_ARGS=--no-group pypi --group cu128
-            platforms: linux/amd64
+            platforms: linux/amd64, linux/arm64
          # - name: docling-project/docling-serve-rocm
          #   build_args: |
          #     UV_SYNC_EXTRA_ARGS=--no-group pypi --group rocm --no-extra flash-attn
--- a/.github/workflows/images.yml
+++ b/.github/workflows/images.yml
@@ -36,7 +36,7 @@ jobs:
          - name: docling-project/docling-serve-cu128
            build_args: |
              UV_SYNC_EXTRA_ARGS=--no-group pypi --group cu128
-            platforms: linux/amd64
+            platforms: linux/amd64, linux/arm64
          # - name: docling-project/docling-serve-rocm
          #   build_args: |
          #     UV_SYNC_EXTRA_ARGS=--no-group pypi --group rocm --no-extra flash-attn
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -98,8 +98,9 @@ cu126 = [
 ]

 cu128 = [
-  "torch>=2.7.1",
+  "torch~=2.7.1",
  "torchvision>=0.22.1",
+  "triton~=3.3.1 ; sys_platform == 'linux'"
 ]

 rocm = [
@@ -146,6 +147,10 @@ torchvision = [
  { index = "pytorch-rocm", group = "rocm", marker = "sys_platform == 'linux'" },
 ]

+triton = [
+  {index = "pytorch-cu128", group = "cu128", marker = "sys_platform == 'linux'" },
+]
+
 pytorch-triton-rocm = [
  { index = "pytorch-rocm", marker = "sys_platform == 'linux'" },
 ]
--- a/uv.lock
+++ b/uv.lock
Author	SHA1	Message	Date
Viktor Kuropiatnyk	86f1f3ebcf	Pinning of dependencies to support cu128 with arm64 Signed-off-by: Viktor Kuropiatnyk <vku@zurich.ibm.com>	2025-10-08 14:31:40 +02:00
Viktor Kuropiatnyk	fe67029592	Removed pr image pushing as it requires a different setup Signed-off-by: Viktor Kuropiatnyk <vku@zurich.ibm.com>	2025-10-08 09:59:31 +02:00
Viktor Kuropiatnyk	b4946369c7	Added arm64 platform for pr, enabled pr image push Signed-off-by: Viktor Kuropiatnyk <vku@zurich.ibm.com>	2025-10-07 16:21:44 +02:00