Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +6 -0
Dockerfile
CHANGED
@@ -4,7 +4,9 @@ ENV MODEL_NAME="llava-1.6-mistral-7b-gguf"
|
|
4 |
ENV DEFAULT_MODEL_FILE="llava-v1.6-mistral-7b.Q3_K_XS.gguf"
|
5 |
ENV MODEL_USER="cjpais"
|
6 |
ENV DEFAULT_MODEL_BRANCH="main"
|
|
|
7 |
ENV MODEL_URL="https://huggingface.co/${MODEL_USER}/${MODEL_NAME}/resolve/${DEFAULT_MODEL_BRANCH}/${DEFAULT_MODEL_FILE}"
|
|
|
8 |
ENV PATH="/usr/local/cuda/bin:$PATH"
|
9 |
|
10 |
RUN apt update && \
|
@@ -36,6 +38,10 @@ WORKDIR /app
|
|
36 |
RUN echo ${MODEL_URL} && \
|
37 |
wget -O /app/${DEFAULT_MODEL_FILE} ${MODEL_URL}
|
38 |
|
|
|
|
|
|
|
|
|
39 |
RUN useradd -m -u 1000 user
|
40 |
|
41 |
RUN mkdir -p /home/user/app && \
|
|
|
4 |
ENV DEFAULT_MODEL_FILE="llava-v1.6-mistral-7b.Q3_K_XS.gguf"
|
5 |
ENV MODEL_USER="cjpais"
|
6 |
ENV DEFAULT_MODEL_BRANCH="main"
|
7 |
+
ENV DEFAULT_CLIP_MODEL_FILE="mmproj-model-f16.gguf"
|
8 |
ENV MODEL_URL="https://huggingface.co/${MODEL_USER}/${MODEL_NAME}/resolve/${DEFAULT_MODEL_BRANCH}/${DEFAULT_MODEL_FILE}"
|
9 |
+
ENV CLIP_MODEL_URL="https://huggingface.co/${MODEL_USER}/${MODEL_NAME}/resolve/${DEFAULT_MODEL_BRANCH}/${DEFAULT_CLIP_MODEL_FILE}"
|
10 |
ENV PATH="/usr/local/cuda/bin:$PATH"
|
11 |
|
12 |
RUN apt update && \
|
|
|
38 |
RUN echo ${MODEL_URL} && \
|
39 |
wget -O /app/${DEFAULT_MODEL_FILE} ${MODEL_URL}
|
40 |
|
41 |
+
RUN echo ${CLIP_MODEL_URL} && \
|
42 |
+
wget -O /app/${DEFAULT_MODEL_FILE} ${CLIP_MODEL_URL}
|
43 |
+
|
44 |
+
|
45 |
RUN useradd -m -u 1000 user
|
46 |
|
47 |
RUN mkdir -p /home/user/app && \
|