Pavishanth68 commited on
Commit
a70cd9d
·
verified ·
1 Parent(s): 4ecb5f3

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +9 -8
Dockerfile CHANGED
@@ -4,9 +4,9 @@ WORKDIR /app
4
 
5
  COPY app.py .
6
 
7
- # Install rust (needed for tokenizers)
8
  RUN apt-get update && apt-get install -y --no-install-recommends \
9
- curl build-essential \
10
  && curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y \
11
  && . $HOME/.cargo/env \
12
  && rm -rf /var/lib/apt/lists/*
@@ -16,15 +16,16 @@ ENV PATH="/root/.cargo/bin:${PATH}"
16
  # Install Python packages
17
  RUN pip install --no-cache-dir \
18
  gradio==4.36.1 \
19
- transformers==4.38.0 \
20
- torch==2.1.0 \
21
- accelerate==0.33.0 \
22
- peft==0.7.1 \
 
23
  sentencepiece==0.2.0 \
24
  protobuf==3.20.3
25
 
26
- # Install tokenizers separately
27
- RUN pip install --no-cache-dir tokenizers==0.15.2
28
 
29
  EXPOSE 7860
30
 
 
4
 
5
  COPY app.py .
6
 
7
+ # Install system dependencies
8
  RUN apt-get update && apt-get install -y --no-install-recommends \
9
+ curl build-essential git \
10
  && curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y \
11
  && . $HOME/.cargo/env \
12
  && rm -rf /var/lib/apt/lists/*
 
16
  # Install Python packages
17
  RUN pip install --no-cache-dir \
18
  gradio==4.36.1 \
19
+ transformers==4.45.2 \
20
+ torch==2.5.1 \
21
+ accelerate==0.34.0 \
22
+ peft==0.13.2 \
23
+ bitsandbytes==0.44.1 \
24
  sentencepiece==0.2.0 \
25
  protobuf==3.20.3
26
 
27
+ # Install tokenizers
28
+ RUN pip install --no-cache-dir tokenizers==0.20.1
29
 
30
  EXPOSE 7860
31