nova commited on
Update Dockerfile
Browse files- 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 |
-
#
|
| 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
|