chenhaodev commited on
Commit
7cbdbbe
1 Parent(s): 6193659

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -4
Dockerfile CHANGED
@@ -3,9 +3,8 @@ FROM python:3.10 as base
3
 
4
  # Set model
5
  ENV MODEL=gguf/Smaug-34B-v0.1-GGUF
6
- ENV QUANT=Q4_K_M
7
- #ENV CHAT_TEMPLATE=chatml
8
- #ENV CHAT_TEMPLATE=llama-2
9
 
10
  # Set up a new user named "user" with user ID 1000
11
  RUN useradd -m -u 1000 user
@@ -22,7 +21,8 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
22
 
23
  # Download model
24
  RUN MODEL_NAME_FILE=$(echo ${MODEL#*/} | tr '[:upper:]' '[:lower:]' | sed 's/-gguf$//') && \
25
- wget https://huggingface.co/gguf/Smaug-34B-v0.1-GGUF/resolve/main/smaug-34b-v0.1.Q4_K_M.gguf -O model.gguf
 
26
 
27
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
28
  COPY --chown=user . $HOME/app
 
3
 
4
  # Set model
5
  ENV MODEL=gguf/Smaug-34B-v0.1-GGUF
6
+ ENV QUANT=Q2_K
7
+ ENV CHAT_TEMPLATE=chatml
 
8
 
9
  # Set up a new user named "user" with user ID 1000
10
  RUN useradd -m -u 1000 user
 
21
 
22
  # Download model
23
  RUN MODEL_NAME_FILE=$(echo ${MODEL#*/} | tr '[:upper:]' '[:lower:]' | sed 's/-gguf$//') && \
24
+ #wget https://huggingface.co/gguf/Smaug-34B-v0.1-GGUF/resolve/main/smaug-34b-v0.1.Q4_K_M.gguf -O model.gguf
25
+ wget https://huggingface.co/gguf/Smaug-34B-v0.1-GGUF/resolve/main/smaug-34b-v0.1.Q2_K.gguf -O model.gguf
26
 
27
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
28
  COPY --chown=user . $HOME/app