From 060102372e630295b67a40007221e5683fcd65ee Mon Sep 17 00:00:00 2001
From: Hanks <hangxu0304@gmail.com>
Date: Fri, 30 May 2025 16:54:27 +0800
Subject: [PATCH] Update release_pypi_after_merge.yml

---
 .github/workflows/release_pypi_after_merge.yml | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/release_pypi_after_merge.yml b/.github/workflows/release_pypi_after_merge.yml
index 6cef8a4b7..e41bc0544 100644
--- a/.github/workflows/release_pypi_after_merge.yml
+++ b/.github/workflows/release_pypi_after_merge.yml
@@ -7,12 +7,14 @@ on:
       - 'version.txt'
     types:
       - closed
-
+permissions:
+  id-token: write
+  contents: read
 jobs:
   build-n-publish:
     if: github.event_name == 'workflow_dispatch' || github.repository == 'hpcaitech/ColossalAI' && github.event.pull_request.merged == true && github.base_ref == 'main'
     name: Build and publish Python 🐍 distributions 📦 to PyPI
-    runs-on: ubuntu-[self-hosted, ubuntu-latest]
+    runs-on: ubuntu-latest
     timeout-minutes: 20
     steps:
     - uses: actions/checkout@v2
@@ -35,7 +37,7 @@ jobs:
   notify:
     name: Notify Lark via webhook
     needs: build-n-publish
-    runs-on: [self-hosted, ubuntu-latest]
+    runs-on: ubuntu-latest
     if: ${{ always() }}
     steps:
       - uses: actions/checkout@v2