SuperSecureHuman commited on
Commit
202c549
1 Parent(s): 9a281c9

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -0
Dockerfile CHANGED
@@ -13,8 +13,11 @@ RUN pip install accelerate pandas numpy
13
  COPY entrypoint.sh /bin/entrypoint.sh
14
  RUN chmod 755 /bin/entrypoint.sh
15
 
 
 
16
  ENV FS_ENABLE_WEB=false
17
  ENV FS_ENABLE_OPENAI_API=true
 
18
 
19
  ENTRYPOINT [ "/bin/bash", "/bin/entrypoint.sh" ]
20
  CMD ["--model-path mistralai/Mistral-7B-Instruct-v0.1"]
 
13
  COPY entrypoint.sh /bin/entrypoint.sh
14
  RUN chmod 755 /bin/entrypoint.sh
15
 
16
+ WORKDIR /app
17
+
18
  ENV FS_ENABLE_WEB=false
19
  ENV FS_ENABLE_OPENAI_API=true
20
+ ENV TRANSFORMERS_CACHE=/app
21
 
22
  ENTRYPOINT [ "/bin/bash", "/bin/entrypoint.sh" ]
23
  CMD ["--model-path mistralai/Mistral-7B-Instruct-v0.1"]