Merge pull request containers#458 from javipolo/custom_instructlab_repo

Allow to use a different instructlab git repository via env vars
mhdawson · May 3, 2024 · d3ba2b6 · d3ba2b6
2 parents e2ead10 + a1ec1ce
commit d3ba2b6
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/training/instructlab/Makefile b/training/instructlab/Makefile
@@ -4,11 +4,14 @@ REGISTRY ?= quay.io
 REGISTRY_ORG ?= ai-lab
 IMAGE_TAG ?= latest
 
+INSTRUCTLAB_GIT_REPO ?= https://github.com/instructlab/instructlab.git
+INSTRUCTLAB_GIT_BRANCH ?= main
+
 .PHONY: instructlab
 instructlab:
 	@mkdir -p ../build
-	git clone https://github.com/instructlab/instructlab.git 2> /dev/null || true
-	(cd instructlab; git pull origin main)
+	git clone $(INSTRUCTLAB_GIT_REPO) instructlab 2> /dev/null || true
+	(cd instructlab; git pull origin $(INSTRUCTLAB_GIT_BRANCH))
 
 .PHONY: nvidia
 nvidia: instructlab