zhengr commited on
Commit
ab1439e
·
verified ·
1 Parent(s): a74136c

Create Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +20 -0
Dockerfile ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # syntax=docker/dockerfile:1
2
+
3
+ FROM ubuntu:22.04
4
+ ARG CHAT_MODEL_FILE
5
+ ARG EMBEDDING_MODEL_FILE
6
+ ARG PROMPT_TEMPLATE
7
+ ENV CHAT_MODEL_FILE=${CHAT_MODEL_FILE}
8
+ ENV EMBEDDING_MODEL_FILE=${EMBEDDING_MODEL_FILE}
9
+ ENV PROMPT_TEMPLATE=${PROMPT_TEMPLATE}
10
+ RUN apt-get update && apt-get install -y curl
11
+ RUN mkdir /models
12
+ COPY $CHAT_MODEL_FILE /models/$CHAT_MODEL_FILE
13
+ COPY $EMBEDDING_MODEL_FILE /models/$EMBEDDING_MODEL_FILE
14
+ RUN curl -sSf https://raw.githubusercontent.com/WasmEdge/WasmEdge/master/utils/install_v2.sh | bash -s -- -v 0.13.5
15
+ RUN curl -LO https://github.com/LlamaEdge/LlamaEdge/releases/latest/download/llama-api-server.wasm
16
+ RUN curl -LO https://github.com/second-state/chatbot-ui/releases/latest/download/chatbot-ui.tar.gz; tar xzf chatbot-ui.tar.gz; rm chatbot-ui.tar.gz
17
+ COPY run.sh .
18
+ RUN chmod +x run.sh
19
+
20
+ ENTRYPOINT ./run.sh $CHAT_MODEL_FILE $EMBEDDING_MODEL_FILE $PROMPT_TEMPLATE "$@"