matthoffner commited on
Commit
ec8373d
·
1 Parent(s): 7a01b2f

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -1,7 +1,7 @@
1
  FROM nvidia/cuda:12.0.0-cudnn8-devel-ubuntu22.04
2
 
3
- ENV MODEL_NAME="Llama-2-13B-chat-GGML"
4
- ENV DEFAULT_MODEL_FILE="llama-2-13b-chat.ggmlv3.q4_K_M.bin"
5
  ENV MODEL_USER="TheBloke"
6
  ENV DEFAULT_MODEL_BRANCH="main"
7
  ENV MODEL_URL="https://huggingface.co/${MODEL_USER}/${MODEL_NAME}/resolve/${DEFAULT_MODEL_BRANCH}/${DEFAULT_MODEL_FILE}"
 
1
  FROM nvidia/cuda:12.0.0-cudnn8-devel-ubuntu22.04
2
 
3
+ ENV MODEL_NAME="StableBeluga-13B-GGML"
4
+ ENV DEFAULT_MODEL_FILE="stablebeluga-7b.ggmlv3.q4_K_M.bin"
5
  ENV MODEL_USER="TheBloke"
6
  ENV DEFAULT_MODEL_BRANCH="main"
7
  ENV MODEL_URL="https://huggingface.co/${MODEL_USER}/${MODEL_NAME}/resolve/${DEFAULT_MODEL_BRANCH}/${DEFAULT_MODEL_FILE}"