Allow to use a different instructlab git repository via env vars

Signed-off-by: Javi Polo <[email protected]>
mhdawson · May 3, 2024 · a1ec1ce · a1ec1ce
1 parent ed3b643
commit a1ec1ce
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