From 4208b98d29b286be7537d9238fef9d11883231e6 Mon Sep 17 00:00:00 2001
From: tastelikefeet <58414341+tastelikefeet@users.noreply.github.com>
Date: Mon, 21 Oct 2024 17:32:42 +0800
Subject: [PATCH] Create docker-image.yml

---
 .github/workflows/docker-image.yml | 31 ++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100644 .github/workflows/docker-image.yml

diff --git a/.github/workflows/docker-image.yml b/.github/workflows/docker-image.yml
new file mode 100644
index 00000000..790ba8f7
--- /dev/null
+++ b/.github/workflows/docker-image.yml
@@ -0,0 +1,31 @@
+name: Build Docker Images
+
+on:
+  workflow_dispatch:
+    inputs:
+      branch:
+        description: 'Branch to build from'
+        required: true
+        default: 'master'
+      image_type:
+        description: 'The image type to build'
+        required: true
+      modelscope_version:
+        description: 'ModelScope version to use'
+        required: true  
+      swift_branch:
+        description: 'SWIFT branch to use'
+        required: true  
+      
+jobs:
+  build:
+    runs-on: modelscope-ci-ecs-us0
+
+    steps:
+    - name: Checkout repository
+      uses: actions/checkout@v4
+      with:
+        ref: ${{ github.event.inputs.branch }}
+
+    - name: Build Docker Image
+      run: python docker/build_image.py --image_type ${{ github.event.inputs.image_type }} --modelscope_branch ${{ github.event.inputs.branch }} --modelscope_version ${{ github.event.inputs.modelscope_version }} --swift_branch ${{ github.event.inputs.swift_branch }}