Skip to content

Commit

Permalink
need to bring in oci artifacts to build
Browse files Browse the repository at this point in the history
Signed-off-by: Ryan Cook <[email protected]>
  • Loading branch information
cooktheryan committed Jun 18, 2024
1 parent 41a8bed commit 00cdd48
Showing 1 changed file with 12 additions and 5 deletions.
17 changes: 12 additions & 5 deletions .github/workflows/training_bootc.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ on:
# paths:
# - .github/workflows/training_bootc.yaml
# - ./training/**

workflow_dispatch:

concurrency:
Expand Down Expand Up @@ -104,12 +104,12 @@ jobs:
sudo rm -rf "$AGENT_TOOLSDIRECTORY"
- uses: actions/[email protected]

- name: Build Image
id: build_image
run: make bootc DRIVER_VERSION=${{ matrix.driver_version }} ARCH=${{ matrix.arch }}
working-directory: ${{ matrix.context }}

- name: Login to Container Registry
if: github.event_name == 'push' && github.ref == 'refs/heads/main'
uses: redhat-actions/[email protected]
Expand Down Expand Up @@ -150,7 +150,7 @@ jobs:
- image_name: amd-bootc
context: training/amd-bootc
arch: amd64
pull-images: quay.io/ai-lab/vllm:latest
pull-images: quay.io/ai-lab/vllm:latest
runs-on: ubuntu-22.04-8-cores
continue-on-error: true
steps:
Expand All @@ -172,7 +172,14 @@ jobs:
id: pull_image
working-directory: ${{ matrix.context }}
run: podman pull ${{ matrix.pull-images }}


- name: generate the local OCI assets
run: |
cd training
make -j vllm
make -j deepspeed
make -j instruct-${{ matrix.arch}}
- name: Build Image
id: build_image
run: make bootc ARCH=${{ matrix.arch }}
Expand Down

0 comments on commit 00cdd48

Please sign in to comment.