optimizing build

redhat-ai-services · Nov 4, 2024 · 1dd1436 · 1dd1436
1 parent 82fc120
commit 1dd1436
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/granite-3.0-2b-instruct/Containerfile b/granite-3.0-2b-instruct/Containerfile
@@ -22,10 +22,10 @@ COPY download_model.py .
 RUN python3 download_model.py
 
 # Final image containing only the essential model files
-FROM registry.access.redhat.com/ubi9/ubi-minimal:9.4
-
-# Copy only the necessary model files from the base image
-COPY --from=base /models /models
+FROM registry.access.redhat.com/ubi9/ubi-micro:9.4
 
 # Set the user to 1001
 USER 1001
+
+# Copy only the necessary model files from the base image
+COPY --from=base /models /models
diff --git a/granite-3.0-2b-instruct/requirements.txt b/granite-3.0-2b-instruct/requirements.txt
@@ -1,3 +1 @@
-torch
-transformers
 huggingface-hub