diff --git a/.github/workflows/training_bootc.yaml b/.github/workflows/training_bootc.yaml index a544015d..d3dca60d 100644 --- a/.github/workflows/training_bootc.yaml +++ b/.github/workflows/training_bootc.yaml @@ -3,24 +3,6 @@ name: Training Bootc image builds on: workflow_dispatch: - pull_request: - branches: - - main - paths: - - .github/workflows/training_bootc.yaml - - 'training/common/**' - - 'training/amd-bootc/**' - - 'training/intel-bootc/**' - - 'training/nvidia-bootc/**' - - 'training/intel-bootc/**' - - 'training/instructlab/**' - - 'training/vllm/**' - # push: - # branches: - # - main - # paths: - # - .github/workflows/model_image_build_push.yaml - concurrency: group: ${{ github.workflow }} cancel-in-progress: false @@ -52,7 +34,7 @@ jobs: mode: start github-token: ${{ secrets.GH_PERSONAL_ACCESS_TOKEN }} ec2-image-id: ami-04cec38d48a5be576 - ec2-instance-type: m7i.8xlarge + ec2-instance-type: m7i.12xlarge subnet-id: subnet-0b1e1d94240813658 security-group-id: sg-055105753f5e8bd83 @@ -118,7 +100,7 @@ jobs: arch: amd64 runs-on: ${{ needs.start-runner.outputs.label }} if: "!contains(github.event.pull_request.labels.*.name, 'hold-tests')" - needs: nvidia-bootc-builder-image + needs: [ nvidia-bootc-builder-image, start-runner ] steps: - uses: actions/checkout@v4.1.7 @@ -126,6 +108,10 @@ jobs: run: | mkdir -p ~/.docker + - name: generate a ssh key - USER SHOULD INJECT THEIR OWN AND REBUILD IF THEY USE THIS IMAGE and overwrite the existing one + run: | + ssh-keygen -t rsa -b 4096 -f ~/.ssh/id_rsa -N "" <<