nova commited on
Commit
c42995b
·
verified ·
1 Parent(s): fa15145

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -3
Dockerfile CHANGED
@@ -4,15 +4,17 @@ WORKDIR /app
4
  RUN apt-get update && apt-get install -y git \
5
  && rm -rf /var/lib/apt/lists/*
6
  # Install Python libs
7
- # optimimum and auto-gptq might be needed for some quantized versions,
8
- # but for standard Phi-3.5 mini:
9
  RUN pip install --no-cache-dir \
10
  torch \
11
  transformers \
12
  accelerate \
13
  gradio \
14
  protobuf \
15
- sentencepiece
 
 
 
16
  # Copy app
17
  COPY . .
18
  # Run
 
4
  RUN apt-get update && apt-get install -y git \
5
  && rm -rf /var/lib/apt/lists/*
6
  # Install Python libs
7
+ # hf_transfer enables faster downloads (Rust-based) to avoid timeouts
 
8
  RUN pip install --no-cache-dir \
9
  torch \
10
  transformers \
11
  accelerate \
12
  gradio \
13
  protobuf \
14
+ sentencepiece \
15
+ hf_transfer
16
+ # Enable HF Transfer
17
+ ENV HF_HUB_ENABLE_HF_TRANSFER=1
18
  # Copy app
19
  COPY . .
20
  # Run