Merge pull request #398 from containers/magma

add model container builder

training/model/Containerfile

@@ -0,0 +1,11 @@
+FROM registry.access.redhat.com/ubi9/ubi
+
+ARG MODEL_REPO=''
+ARG MODEL_NAME=''
+ARG MODEL_PATH=''
+
+RUN dnf install -y python3-pip && python3 -m pip install huggingface_hub
+RUN mkdir -p "${MODEL_PATH}" \
+    && echo from huggingface_hub import snapshot_download > /root/hf_download \
+    && echo snapshot_download\(repo_id=\'${MODEL_REPO}\', local_dir=\'${MODEL_PATH}\', local_dir_use_symlinks=False\) >> /root/hf_download \
+    && python3 /root/hf_download

training/model/Makefile

@@ -0,0 +1,22 @@
+FROM ?= 
+
+REGISTRY ?= quay.io
+REGISTRY_ORG ?= ai-lab
+IMAGE_NAME ?= granite-7b-lab
+IMAGE_TAG ?= latest
+
+CONTAINER_TOOL ?= podman
+CONTAINER_TOOL_EXTRA_ARGS ?=
+
+MODEL_REPO ?= ibm/granite-7b-base
+MODEL_PATH ?= /usr/share/ai-model
+
+.PHONY: image
+image:
+	"${CONTAINER_TOOL}" build \
+		--file Containerfile \
+		--tag "${REGISTRY}/${REGISTRY_ORG}/${IMAGE_NAME}:${IMAGE_TAG}" \
+		$(FROM:%=--build-arg BASEIMAGE=%) \
+		$(MODEL_REPO:%=--build-arg MODEL_REPO=%) \
+		$(MODEL_PATH:%=--build-arg MODEL_PATH=%) \
+		${CONTAINER_TOOL_EXTRA_ARGS}