Spaces:
Runtime error
Runtime error
Abhishek commited on
Commit ·
0ebec80
1
Parent(s): 08889f3
chore: update model paths to point to /data/
Browse files- Dockerfile +3 -3
Dockerfile
CHANGED
|
@@ -5,7 +5,7 @@ ENV DEBIAN_FRONTEND=noninteractive \
|
|
| 5 |
PORT=7860 \
|
| 6 |
HOST=0.0.0.0 \
|
| 7 |
GRADIO_SERVER_NAME=0.0.0.0 \
|
| 8 |
-
P5_FLUX_MODEL_DIR=/
|
| 9 |
|
| 10 |
RUN apt-get update \
|
| 11 |
&& apt-get install -y --no-install-recommends git ca-certificates libcurl4 libgomp1 libstdc++6 \
|
|
@@ -28,8 +28,8 @@ ENV HF_TOKEN=${HF_TOKEN}
|
|
| 28 |
RUN pip install huggingface_hub hf_transfer
|
| 29 |
RUN --mount=type=secret,id=HF_TOKEN,mode=0444,required=false \
|
| 30 |
if [ -f /run/secrets/HF_TOKEN ]; then export HF_TOKEN=$(cat /run/secrets/HF_TOKEN); fi && \
|
| 31 |
-
python scripts/download_model.py black-forest-labs/FLUX.2-klein-base-4B
|
| 32 |
-
python scripts/download_model.py openbmb/MiniCPM5-1B
|
| 33 |
|
| 34 |
EXPOSE 7860
|
| 35 |
CMD ["python", "app.py"]
|
|
|
|
| 5 |
PORT=7860 \
|
| 6 |
HOST=0.0.0.0 \
|
| 7 |
GRADIO_SERVER_NAME=0.0.0.0 \
|
| 8 |
+
P5_FLUX_MODEL_DIR=/data/FLUX.2-klein-base-4B
|
| 9 |
|
| 10 |
RUN apt-get update \
|
| 11 |
&& apt-get install -y --no-install-recommends git ca-certificates libcurl4 libgomp1 libstdc++6 \
|
|
|
|
| 28 |
RUN pip install huggingface_hub hf_transfer
|
| 29 |
RUN --mount=type=secret,id=HF_TOKEN,mode=0444,required=false \
|
| 30 |
if [ -f /run/secrets/HF_TOKEN ]; then export HF_TOKEN=$(cat /run/secrets/HF_TOKEN); fi && \
|
| 31 |
+
python scripts/download_model.py black-forest-labs/FLUX.2-klein-base-4B /data/FLUX.2-klein-base-4B && \
|
| 32 |
+
python scripts/download_model.py openbmb/MiniCPM5-1B /data/MiniCPM5-1B
|
| 33 |
|
| 34 |
EXPOSE 7860
|
| 35 |
CMD ["python", "app.py"]
|