Install FLUX dependencies in isolated venv
This commit is contained in:
@@ -5,7 +5,8 @@ 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 pip install --no-cache-dir \
|
RUN python -m venv --system-site-packages /opt/flux-venv && \
|
||||||
|
/opt/flux-venv/bin/pip install --no-cache-dir \
|
||||||
"diffusers==${DIFFUSERS_VERSION}" \
|
"diffusers==${DIFFUSERS_VERSION}" \
|
||||||
"transformers==${TRANSFORMERS_VERSION}" \
|
"transformers==${TRANSFORMERS_VERSION}" \
|
||||||
"accelerate==${ACCELERATE_VERSION}" \
|
"accelerate==${ACCELERATE_VERSION}" \
|
||||||
@@ -15,4 +16,4 @@ RUN pip install --no-cache-dir \
|
|||||||
|
|
||||||
COPY flux_worker.py /app/flux_worker.py
|
COPY flux_worker.py /app/flux_worker.py
|
||||||
USER 10002:10002
|
USER 10002:10002
|
||||||
ENTRYPOINT ["python", "/app/flux_worker.py"]
|
ENTRYPOINT ["/opt/flux-venv/bin/python", "/app/flux_worker.py"]
|
||||||
|
|||||||
Reference in New Issue
Block a user