Add venv support to FLUX worker image
This commit is contained in:
@@ -5,7 +5,9 @@ ARG TRANSFORMERS_VERSION=5.15.1
|
|||||||
ARG ACCELERATE_VERSION=1.14.0
|
ARG ACCELERATE_VERSION=1.14.0
|
||||||
ARG HF_HUB_VERSION=1.28.0
|
ARG HF_HUB_VERSION=1.28.0
|
||||||
|
|
||||||
RUN python -m venv --system-site-packages /opt/flux-venv && \
|
RUN apt-get update && apt-get install -y --no-install-recommends python3.12-venv && \
|
||||||
|
rm -rf /var/lib/apt/lists/* && \
|
||||||
|
python -m venv --system-site-packages /opt/flux-venv && \
|
||||||
/opt/flux-venv/bin/pip install --no-cache-dir \
|
/opt/flux-venv/bin/pip install --no-cache-dir \
|
||||||
"diffusers==${DIFFUSERS_VERSION}" \
|
"diffusers==${DIFFUSERS_VERSION}" \
|
||||||
"transformers==${TRANSFORMERS_VERSION}" \
|
"transformers==${TRANSFORMERS_VERSION}" \
|
||||||
|
|||||||
Reference in New Issue
Block a user