Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +3 -3
Dockerfile
CHANGED
|
@@ -1,8 +1,8 @@
|
|
| 1 |
FROM nvidia/cuda:11.0.3-base-ubuntu20.04
|
| 2 |
|
| 3 |
-
ENV MODEL_NAME="
|
| 4 |
-
ENV DEFAULT_MODEL_FILE="
|
| 5 |
-
ENV MODEL_USER="
|
| 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"
|
|
|
|
| 1 |
FROM nvidia/cuda:11.0.3-base-ubuntu20.04
|
| 2 |
|
| 3 |
+
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"
|