better download script

Signed-off-by: Costin Gamenț <[email protected]>
containers · Apr 30, 2024 · 4490863 · 4490863
1 parent c3197ce
commit 4490863
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/training/model/Containerfile b/training/model/Containerfile
@@ -1,8 +1,11 @@
-FROM quay.io/centos-bootc/centos-bootc:stream9
+FROM registry.access.redhat.com/ubi9/ubi
 
 ARG MODEL_REPO=''
+ARG MODEL_NAME=''
 ARG MODEL_PATH=''
 
-RUN dnf install -y python3-pip && python -m pip install huggingface_hub
+RUN dnf install -y python3-pip && python3 -m pip install huggingface_hub
 RUN mkdir -p "${MODEL_PATH}" \
-    && echo from huggingface_hub import snapshot_download\;snapshot_download\(repo_id=\'${MODEL_REPO}\', local_dir=\'${MODEL_PATH}\', local_dir_use_symlinks=False\) | python
+    && echo from huggingface_hub import snapshot_download > /root/hf_download \
+    && echo snapshot_download\(repo_id=\'${MODEL_REPO}\', local_dir=\'${MODEL_PATH}\', local_dir_use_symlinks=False\) >> /root/hf_download \
+    && python3 /root/hf_download