first backup of charts
This commit is contained in:
44
vllm/vllm-app/templates/model-download-job.yaml
Normal file
44
vllm/vllm-app/templates/model-download-job.yaml
Normal file
@ -0,0 +1,44 @@
|
||||
{{- if .Values.model.download.enabled }}
|
||||
apiVersion: batch/v1
|
||||
kind: Job
|
||||
metadata:
|
||||
name: {{ .Release.Name }}-download-model
|
||||
annotations:
|
||||
"helm.sh/hook": pre-install,pre-upgrade # 在安装/升级前执行
|
||||
"helm.sh/hook-weight": "-10" # 优先执行
|
||||
"helm.sh/hook-delete-policy": hook-succeeded
|
||||
spec:
|
||||
template:
|
||||
spec:
|
||||
restartPolicy: OnFailure
|
||||
containers:
|
||||
- name: downloader
|
||||
image: {{ .Values.model.download.image }}
|
||||
env:
|
||||
- name: HF_ENDPOINT
|
||||
value: https://hf-mirror.com
|
||||
- name: HUGGING_FACE_HUB_TOKEN
|
||||
value: {{ .Values.model.huggingfaceToken }}
|
||||
command:
|
||||
- sh
|
||||
- -c
|
||||
- |
|
||||
DEST_DIR="{{ .Values.model.localMountPath }}/{{ .Values.model.huggingfaceName }}"
|
||||
if [ -d "$DEST_DIR" ]; then
|
||||
echo "Model already exists at $DEST_DIR"
|
||||
else
|
||||
echo "Downloading model {{ .Values.model.huggingfaceName }} to $DEST_DIR"
|
||||
wget https://hf-mirror.com/hfd/hfd.sh
|
||||
chmod a+x hfd.sh
|
||||
apt install aria2 -y
|
||||
./hfd.sh {{ .Values.model.huggingfaceName }} --local-dir "$DEST_DIR"
|
||||
# huggingface-cli download {{ .Values.model.huggingfaceName }} --local-dir "$DEST_DIR"
|
||||
fi
|
||||
volumeMounts:
|
||||
- name: model-storage
|
||||
mountPath: {{ .Values.model.localMountPath }}
|
||||
volumes:
|
||||
- name: model-storage
|
||||
persistentVolumeClaim:
|
||||
claimName: nfs-pvc-model # 复用之前的 PVC
|
||||
{{- end }}
|
||||
Reference in New Issue
Block a user