Intermediary model container

Signed-off-by: Costin Gamenț <[email protected]>

training/model/Containerfile

@@ -0,0 +1,8 @@
+FROM quay.io/centos-bootc/centos-bootc:stream9
+
+ARG MODEL_REPO=https://huggingface.co/ibm/granite-7b-base
+ARG MODEL_PATH=/usr/share/ai-model
+
+RUN dnf install -y git-lfs
+RUN mkdir -p "${MODEL_PATH}"
+RUN git clone --progress --verbose "${MODEL_REPO}" "${MODEL_PATH}"

training/model/Makefile

@@ -0,0 +1,22 @@
+FROM ?= 
+
+REGISTRY ?= quay.io
+REGISTRY_ORG ?= ai-lab
+IMAGE_NAME ?= granite-model
+IMAGE_TAG ?= latest
+
+CONTAINER_TOOL ?= podman
+CONTAINER_TOOL_EXTRA_ARGS ?=
+
+MODEL_REPO ?=
+MODEL_PATH ?= 
+
+.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}