update model name

redhat-ai-services · Nov 4, 2024 · 32e58d0 · 32e58d0
1 parent 2cfec2e
commit 32e58d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/granite-3.0-2b-instruct/Containerfile b/granite-3.0-2b-instruct/Containerfile
@@ -5,7 +5,7 @@ FROM registry.access.redhat.com/ubi9/ubi-minimal:9.4 as base
 ARG HF_TOKEN
 
 # The model repo to download
-ENV MODEL_REPO="mistralai/Mistral-7B-Instruct-v0.3"
+ENV MODEL_REPO="ibm-granite/granite-3.0-2b-instruct"
 
 # Install necessary Python dependencies
 RUN microdnf -y install git git-lfs python3-pip && \