gfjiogopdfgdfs commited on
Commit
b9e6e86
·
verified ·
1 Parent(s): 607c143

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +13 -20
Dockerfile CHANGED
@@ -1,4 +1,16 @@
1
  FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
 
 
 
 
 
 
 
 
 
 
 
 
2
 
3
  # Install necessary packages
4
  RUN apt update && apt install git build-essential libopenblas-dev wget python3-pip -y
@@ -10,26 +22,7 @@ ARG MODEL_PART_B
10
  ARG MODEL_NAME
11
  ARG ADDITIONAL
12
  RUN git clone https://github.com/theroyallab/tabbyAPI
13
- WORKDIR /tabbyAPI
14
  RUN pip install -r requirements.txt -q
15
  RUN pip install huggingface-hub -q
16
- RUN echo "network:" >> config.yml && \
17
- echo " host: 127.0.0.1" >> config.yml && \
18
- echo " port: 5000" >> config.yml && \
19
- echo " disable_auth: False" >> config.yml && \
20
- echo "" >> config.yml && \
21
- echo "logging:" >> config.yml && \
22
- echo " prompt: False" >> config.yml && \
23
- echo " generation_params: False" >> config.yml && \
24
- echo "" >> config.yml && \
25
- echo "sampling:" >> config.yml && \
26
- echo " override_preset: null" >> config.yml && \
27
- echo "" >> config.yml && \
28
- echo "developer:" >> config.yml && \
29
- echo " unsafe_launch: False" >> config.yml && \
30
- echo "" >> config.yml && \
31
- echo "model:" >> config.yml && \
32
- echo " model_dir: models" >> config.yml && \
33
- echo " model_name: null" >> config.yml && \
34
- echo " use_dummy_models: False" >> config.yml
35
  CMD ["python3", "main.py"]
 
1
  FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
2
+ # Set up a new user named "user" with user ID 1000
3
+ RUN useradd -m -u 1000 user
4
+
5
+ # Switch to the "user" user
6
+ USER user
7
+
8
+ # Set home to the user's home directory
9
+ ENV HOME=/home/user \
10
+ PATH=/home/user/.local/bin:$PATH
11
+
12
+ # Set the working directory to the user's home directory
13
+ WORKDIR $HOME/app
14
 
15
  # Install necessary packages
16
  RUN apt update && apt install git build-essential libopenblas-dev wget python3-pip -y
 
22
  ARG MODEL_NAME
23
  ARG ADDITIONAL
24
  RUN git clone https://github.com/theroyallab/tabbyAPI
25
+ WORKDIR $HOME/app
26
  RUN pip install -r requirements.txt -q
27
  RUN pip install huggingface-hub -q
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
28
  CMD ["python3", "main.py"]