Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +9 -8
Dockerfile
CHANGED
|
@@ -4,9 +4,9 @@ WORKDIR /app
|
|
| 4 |
|
| 5 |
COPY app.py .
|
| 6 |
|
| 7 |
-
# Install
|
| 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.
|
| 20 |
-
torch==2.
|
| 21 |
-
accelerate==0.
|
| 22 |
-
peft==0.
|
|
|
|
| 23 |
sentencepiece==0.2.0 \
|
| 24 |
protobuf==3.20.3
|
| 25 |
|
| 26 |
-
# Install tokenizers
|
| 27 |
-
RUN pip install --no-cache-dir tokenizers==0.
|
| 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 |
|