From 1bb7e10ba5735ba0cde623f23a2f68d5cce96ab3 Mon Sep 17 00:00:00 2001 From: Dushyant Behl Date: Fri, 7 Nov 2025 08:42:05 +0530 Subject: [PATCH 1/2] Add free up disk space to gh runners Signed-off-by: Dushyant Behl --- .github/workflows/build-and-publish.yaml | 2 ++ .github/workflows/coverage.yaml | 2 ++ .github/workflows/format.yml | 2 ++ 3 files changed, 6 insertions(+) diff --git a/.github/workflows/build-and-publish.yaml b/.github/workflows/build-and-publish.yaml index 208f6b2088..f728d51a33 100644 --- a/.github/workflows/build-and-publish.yaml +++ b/.github/workflows/build-and-publish.yaml @@ -35,6 +35,8 @@ jobs: steps: - uses: actions/checkout@v4 + - name: "Free up disk space" + uses: ./.github/actions/free-up-disk-space - name: Set up Python ${{ matrix.python-version.setup }} uses: actions/setup-python@v4 with: diff --git a/.github/workflows/coverage.yaml b/.github/workflows/coverage.yaml index 75415a749b..df1d046e97 100644 --- a/.github/workflows/coverage.yaml +++ b/.github/workflows/coverage.yaml @@ -10,6 +10,8 @@ jobs: runs-on: ubuntu-latest steps: - uses: actions/checkout@v4 + - name: "Free up disk space" + uses: ./.github/actions/free-up-disk-space - name: Set up Python 3.12 uses: actions/setup-python@v4 with: diff --git a/.github/workflows/format.yml b/.github/workflows/format.yml index 777a3853bb..3122efb0e3 100644 --- a/.github/workflows/format.yml +++ b/.github/workflows/format.yml @@ -25,6 +25,8 @@ jobs: runs-on: ubuntu-latest steps: - uses: actions/checkout@v4 + - name: "Free up disk space" + uses: ./.github/actions/free-up-disk-space - name: Set up Python 3.12 uses: actions/setup-python@v4 with: From f84ebb0a5277198ae0060782435b3e2183cf7bc9 Mon Sep 17 00:00:00 2001 From: Dushyant Behl Date: Fri, 7 Nov 2025 09:16:20 +0530 Subject: [PATCH 2/2] Build flash attn without cache to avoid problems Signed-off-by: Dushyant Behl --- build/Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/build/Dockerfile b/build/Dockerfile index 836373b547..4fec1bb52e 100644 --- a/build/Dockerfile +++ b/build/Dockerfile @@ -156,9 +156,10 @@ ENV PIP_NO_BINARY=mamba-ssm,mamba_ssm RUN --mount=type=cache,target=/home/${USER}/.cache/pip,uid=${USER_UID} \ python -m pip install --user wheel && \ python -m pip install --user "$(head bdist_name)" && \ - python -m pip install --user "$(head bdist_name)[flash-attn]" && \ python -m pip install --user --no-build-isolation "$(head bdist_name)[mamba]" +RUN python -m pip install --user --no-build-isolation "$(head bdist_name)[flash-attn]" + # fms_acceleration_peft = PEFT-training, e.g., 4bit QLoRA # fms_acceleration_foak = Fused LoRA and triton kernels # fms_acceleration_aadp = Padding-Free Flash Attention Computation