Pavishanth68 commited on
Commit
c7d82d4
·
verified ·
1 Parent(s): ab7cf07

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -4
Dockerfile CHANGED
@@ -13,17 +13,17 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
13
 
14
  ENV PATH="/root/.cargo/bin:${PATH}"
15
 
16
- # Install Python packages
17
  RUN pip install --no-cache-dir \
18
  gradio==4.36.1 \
19
- transformers==4.44.0 \
20
  torch==2.1.0 \
21
  accelerate==0.33.0 \
22
  sentencepiece==0.2.0 \
23
  protobuf==3.20.3
24
 
25
- # Install tokenizers separately to ensure it builds
26
- RUN pip install --no-cache-dir tokenizers==0.19.1
27
 
28
  EXPOSE 7860
29
 
 
13
 
14
  ENV PATH="/root/.cargo/bin:${PATH}"
15
 
16
+ # Install Python packages with OLDER transformers version
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
  sentencepiece==0.2.0 \
23
  protobuf==3.20.3
24
 
25
+ # Install tokenizers separately
26
+ RUN pip install --no-cache-dir tokenizers==0.15.2
27
 
28
  EXPOSE 7860
29