phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Do not cache pip (#349)

Browse source 
Pip has a 3.3GB cache of torch and friends. Do not keep this in the image.
This commit is contained in:
Steve Grubb 2024-10-31 12:52:40 -04:00 • committed by GitHub
parent 3b1917d5ea
commit f04b566c5c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
llama_stack/distribution/build_container.sh
View file

@ -77,9 +77,9 @@ if [ -n "$LLAMA_STACK_DIR" ]; then
# Install in editable format. We will mount the source code into the container
# so that changes will be reflected in the container without having to do a
# rebuild. This is just for development convenience.
add_to_docker "RUN pip install -e $stack_mount"
add_to_docker "RUN pip install --no-cache -e $stack_mount"
else
add_to_docker "RUN pip install llama-stack"
add_to_docker "RUN pip install --no-cache llama-stack"
fi
if [ -n "$LLAMA_MODELS_DIR" ]; then
@ -90,19 +90,19 @@ if [ -n "$LLAMA_MODELS_DIR" ]; then
add_to_docker <<EOF
RUN pip uninstall -y llama-models
RUN pip install $models_mount
RUN pip install --no-cache $models_mount
EOF
fi
if [ -n "$pip_dependencies" ]; then
add_to_docker "RUN pip install $pip_dependencies"
add_to_docker "RUN pip install --no-cache $pip_dependencies"
fi
if [ -n "$special_pip_deps" ]; then
IFS='#' read -ra parts <<<"$special_pip_deps"
for part in "${parts[@]}"; do
add_to_docker "RUN pip install $part"
add_to_docker "RUN pip install --no-cache $part"
done
fi