Merge pull request #683 from javipolo/allow_instructlab_override

Allow INSTRUCTLAB_IMAGE value override
containers · Jul 21, 2024 · ca35deb · ca35deb
2 parents 7aee0bb + 15a8da9
commit ca35deb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/training/common/Makefile.common b/training/common/Makefile.common
@@ -36,7 +36,7 @@ ARCH ?=
 DRIVER_VERSION ?=
 KERNEL_VERSION ?=
 
-INSTRUCTLAB_IMAGE = $(REGISTRY)/$(REGISTRY_ORG)/instructlab-$(HARDWARE):$(IMAGE_TAG)
+INSTRUCTLAB_IMAGE ?= $(REGISTRY)/$(REGISTRY_ORG)/instructlab-$(HARDWARE):$(IMAGE_TAG)
 WRAPPER = $(CURDIR)/../ilab-wrapper/ilab
 QLORA_WRAPPER = $(CURDIR)/../ilab-wrapper/ilab-qlora
 TRAIN_WRAPPER =  $(CURDIR)/../ilab-wrapper/ilab-training-launcher