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 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}" \
|
||||
"transformers==${TRANSFORMERS_VERSION}" \
|
||||
"accelerate==${ACCELERATE_VERSION}" \
|
||||
@@ -15,4 +16,4 @@ RUN pip install --no-cache-dir \
|
||||
|
||||
COPY flux_worker.py /app/flux_worker.py
|
||||
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