SuperSecureHuman commited on
Commit
9a281c9
·
1 Parent(s): 9208e62

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -7
Dockerfile CHANGED
@@ -1,21 +1,20 @@
1
- # Use Ubuntu 22.04 as the base image
2
  FROM ubuntu:22.04
3
 
4
- # Install Python 3 and pip
5
  RUN apt-get update && \
6
  apt-get install -y python3 python3-pip && \
7
  apt-get clean && \
8
  rm -rf /var/lib/apt/lists/*
9
 
10
- # Upgrade pip and install Python packages
11
  RUN pip3 install --upgrade pip
12
  RUN pip3 install transformers
13
  RUN pip3 install "fschat[model_worker,webui]"
14
  RUN pip install accelerate pandas numpy
15
 
 
 
16
 
17
- # Add the entrypoint script
18
- ADD entrypoint.sh /entrypoint.sh
19
 
20
- # Set the script as the entry point
21
- ENTRYPOINT ["/entrypoint.sh"]
 
 
1
  FROM ubuntu:22.04
2
 
 
3
  RUN apt-get update && \
4
  apt-get install -y python3 python3-pip && \
5
  apt-get clean && \
6
  rm -rf /var/lib/apt/lists/*
7
 
 
8
  RUN pip3 install --upgrade pip
9
  RUN pip3 install transformers
10
  RUN pip3 install "fschat[model_worker,webui]"
11
  RUN pip install accelerate pandas numpy
12
 
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"]