From eef473225db1fcf75ab34cda8be70869ca1f99b7 Mon Sep 17 00:00:00 2001
From: Junda Su <90978028+Davids048@users.noreply.github.com>
Date: Wed, 20 May 2026 13:50:02 -0700
Subject: [PATCH] [ci] Add Dreamverse Docker image workflow (#1369)

---
 .github/workflows/_template-build-image.yml | 10 +++++++---
 .github/workflows/infra-build-image.yml     | 15 +++++++++++++++
 2 files changed, 22 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/_template-build-image.yml b/.github/workflows/_template-build-image.yml
index 3eb7bc84f..f0c0961b4 100644
--- a/.github/workflows/_template-build-image.yml
+++ b/.github/workflows/_template-build-image.yml
@@ -12,6 +12,10 @@ on:
       tag_suffix:
         required: true
         type: string
+      image_name:
+        required: false
+        type: string
+        default: fastvideo-dev
 
 jobs:
   build-and-push:
@@ -85,7 +89,7 @@ jobs:
         id: meta
         uses: docker/metadata-action@v5
         with:
-          images: ghcr.io/${{ github.repository }}/fastvideo-dev
+          images: ghcr.io/${{ github.repository }}/${{ inputs.image_name }}
           tags: ${{ steps.prepare-tags.outputs.tags }}
       
       - name: Build and push Docker image
@@ -102,5 +106,5 @@ jobs:
       
       - name: Success message
         run: |
-          echo "✅ Python ${{ inputs.python_version }} image successfully built and pushed to ghcr.io/${{ github.repository }}/fastvideo-dev:${{ inputs.tag_suffix }}-latest"
-          echo "To run tests with this image, manually trigger the 'Run Tests' workflow." 
\ No newline at end of file
+          echo "✅ Python ${{ inputs.python_version }} image successfully built and pushed to ghcr.io/${{ github.repository }}/${{ inputs.image_name }}:${{ inputs.tag_suffix }}-latest"
+          echo "To run tests with this image, manually trigger the 'Run Tests' workflow."
diff --git a/.github/workflows/infra-build-image.yml b/.github/workflows/infra-build-image.yml
index 145048fb9..43f81d35e 100644
--- a/.github/workflows/infra-build-image.yml
+++ b/.github/workflows/infra-build-image.yml
@@ -23,6 +23,11 @@ on:
         required: false
         default: false
         type: boolean
+      dreamverse_cuda_12_9:
+        description: 'Build Dreamverse CUDA 12.9 image'
+        required: false
+        default: false
+        type: boolean
 
 
 permissions:
@@ -65,3 +70,13 @@ jobs:
       dockerfile_path: docker/Dockerfile.python3.12.cuda12.9.1
       tag_suffix: py3.12-cuda12.9.1
     secrets: inherit
+
+  build-dreamverse-cuda-12-9:
+    if: ${{ github.event.inputs.dreamverse_cuda_12_9 == 'true' }}
+    uses: ./.github/workflows/_template-build-image.yml
+    with:
+      python_version: '3.12'
+      dockerfile_path: apps/dreamverse/docker/Dockerfile
+      tag_suffix: dreamverse-cuda12.9.1
+      image_name: dreamverse
+    secrets: inherit