Sergidev commited on
Commit
29ca3eb
·
verified ·
1 Parent(s): 4a07f1a

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +9 -6
Dockerfile CHANGED
@@ -5,19 +5,22 @@ RUN apt-get update && apt-get install -y \
5
  git \
6
  python3.10 \
7
  python3-pip \
 
8
  && rm -rf /var/lib/apt/lists/*
9
 
10
- # Set working directory
11
  WORKDIR /app
12
 
13
- # Install Python packages
14
- COPY requirements.txt .
15
- RUN pip3 install --no-cache-dir -r requirements.txt
16
 
17
- # Install custom FastChat version
18
  RUN git clone -b self-lengthen https://github.com/quanshr/FastChat.git && \
19
  cd FastChat && \
20
- pip3 install -e ".[model_worker,webui]"
 
 
 
 
21
 
22
  # Copy project files
23
  COPY . .
 
5
  git \
6
  python3.10 \
7
  python3-pip \
8
+ wget \
9
  && rm -rf /var/lib/apt/lists/*
10
 
 
11
  WORKDIR /app
12
 
13
+ # Install PyTorch first
14
+ RUN pip3 install --no-cache-dir torch==2.4.0
 
15
 
16
+ # Install FastChat from source without editable mode
17
  RUN git clone -b self-lengthen https://github.com/quanshr/FastChat.git && \
18
  cd FastChat && \
19
+ pip3 install ".[model_worker,webui]"
20
+
21
+ # Install other dependencies
22
+ COPY requirements.txt .
23
+ RUN pip3 install --no-cache-dir -r requirements.txt
24
 
25
  # Copy project files
26
  COPY . .