diff --git a/.github/workflows/integration-tests.yml b/.github/workflows/integration-tests.yml
index da41e2185..66da7f592 100644
--- a/.github/workflows/integration-tests.yml
+++ b/.github/workflows/integration-tests.yml
@@ -29,6 +29,12 @@ jobs:
       fail-fast: false # we want to run all tests regardless of failure
 
     steps:
+      # Huggingface trainer provider will download a model to train and save
+      # checkpoints, so need more space than other flows for training.
+      - name: Free disk space
+        if: ${{ matrix.test-type == 'post_training' }}
+        uses: jlumbroso/free-disk-space@54081f138730dfa15788a46383842cd2f914a1be # v1.3.1
+
       - name: Checkout repository
         uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
 
@@ -85,6 +91,7 @@ jobs:
             echo "Ollama health check failed"
             exit 1
           fi
+
       - name: Check Storage and Memory Available Before Tests
         if: ${{ always() }}
         run: |