Docker fix (#874)

flexflow · Jul 19, 2023 · 190c600 · 190c600
1 parent 8b18d11
commit 190c600
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
@@ -44,7 +44,7 @@ jobs:
     env:
       FF_GPU_BACKEND: ${{ matrix.gpu_backend }}
       cuda_version: ${{ matrix.cuda_version }}
-      branch_name: ${GITHUB_REF#refs/heads/}
+      branch_name: ${{ github.head_ref || github.ref_name }}
     steps:
       - name: Checkout Git Repository
         uses: actions/checkout@v3