dongsiqie commited on
Commit
b65b029
1 Parent(s): 4180d7e

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -0
Dockerfile CHANGED
@@ -4,6 +4,11 @@ RUN apt install git
4
  RUN git clone https://github.com/Hansimov/hf-llm-api.git /app
5
  WORKDIR "/app"
6
  RUN mkdir /.cache && chmod 777 /.cache
 
 
 
 
7
  RUN pip install -r requirements.txt
 
8
  EXPOSE 23333
9
  CMD ["python", "-m", "apis.chat_api"]
 
4
  RUN git clone https://github.com/Hansimov/hf-llm-api.git /app
5
  WORKDIR "/app"
6
  RUN mkdir /.cache && chmod 777 /.cache
7
+ RUN pip install torch
8
+ RUN pip install tensorflow>=2.0
9
+ RUN pip install flax
10
+
11
  RUN pip install -r requirements.txt
12
+
13
  EXPOSE 23333
14
  CMD ["python", "-m", "apis.chat_api"]