File tree Expand file tree Collapse file tree 1 file changed +12
-1
lines changed Expand file tree Collapse file tree 1 file changed +12
-1
lines changed Original file line number Diff line number Diff line change @@ -99,4 +99,15 @@ CMD ["invokeai-web"]
99
99
COPY --link --from=web-builder /build/dist ${INVOKEAI_SRC}/invokeai/frontend/web/dist
100
100
101
101
# add sources last to minimize image changes on code changes
102
- COPY invokeai ${INVOKEAI_SRC}/invokeai
102
+ COPY invokeai ${INVOKEAI_SRC}/invokeai
103
+
104
+ # this should not increase image size because we've already installed dependencies
105
+ # in a previous layer
106
+ RUN --mount=type=cache,target=/root/.cache/uv \
107
+ --mount=type=bind,source=pyproject.toml,target=pyproject.toml \
108
+ --mount=type=bind,source=uv.lock,target=uv.lock \
109
+ if [ "$TARGETPLATFORM" = "linux/arm64" ] || [ "$GPU_DRIVER" = "cpu" ]; then UV_INDEX="https://download.pytorch.org/whl/cpu" ; \
110
+ elif [ "$GPU_DRIVER" = "rocm" ]; then UV_INDEX="https://download.pytorch.org/whl/rocm6.2" ; \
111
+ fi && \
112
+ uv pip install -e .
113
+
You can’t perform that action at this time.
0 commit comments