Skip to content

Commit

Permalink
Change source from ilab wrapper
Browse files Browse the repository at this point in the history
Signed-off-by: Enrique Belarte Luque <ebelarte@redhat.com>
  • Loading branch information
enriquebelarte committed Jun 21, 2024
1 parent 5ff1da8 commit 528ae44
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 4 deletions.
5 changes: 3 additions & 2 deletions training/amd-bootc/Containerfile
Original file line number Diff line number Diff line change
Expand Up @@ -22,9 +22,10 @@ RUN dnf install -y \
# Setup /usr/lib/containers/storage as an additional store for images.
# Remove once the base images have this set by default.
RUN sed -i -e '/additionalimage.*/a "/usr/lib/containers/storage",' \
/etc/containers/storage.conf && \
cp /run/.input/ilab /usr/local/bin/ilab
/etc/containers/storage.conf

COPY ../ilab-wrapper/ilab /usr/local/bin/ilab

ARG INSTRUCTLAB_IMAGE="quay.io/ai-lab/instructlab-amd:latest"

ARG SSHPUBKEY
Expand Down
5 changes: 3 additions & 2 deletions training/nvidia-bootc/Containerfile
Original file line number Diff line number Diff line change
Expand Up @@ -156,8 +156,9 @@ fi
# Also make sure not to duplicate if a base image already has it specified.
RUN grep -q /usr/lib/containers/storage /etc/containers/storage.conf || \
sed -i -e '/additionalimage.*/a "/usr/lib/containers/storage",' \
/etc/containers/storage.conf && \
cp /run/.input/ilab /usr/local/bin/ilab
/etc/containers/storage.conf

COPY ../ilab-wrapper/ilab /usr/local/bin/ilab

ARG INSTRUCTLAB_IMAGE="quay.io/ai-lab/instructlab-nvidia:latest"
ARG GPU_COUNT_COMMAND="nvidia-ctk --quiet cdi list | grep -P nvidia.com/gpu='\\\\d+' | wc -l"
Expand Down

0 comments on commit 528ae44

Please sign in to comment.