Skip to content

GitLab

Unverified Commit 3e3d4692 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[Dockerfile] remove uv from docker jax tpu (#7115) 

remove uv from docker jax tpu
parent d71ecad8
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
docker/diffusers-flax-tpu/Dockerfile
View file @ 3e3d4692
...@@ -24,7 +24,7 @@ ENV PATH="/opt/venv/bin:$PATH" ...@@ -24,7 +24,7 @@ ENV PATH="/opt/venv/bin:$PATH"
# pre-install the heavy dependencies (these can later be overridden by the deps from setup.py) # pre-install the heavy dependencies (these can later be overridden by the deps from setup.py)
# follow the instructions here: https://cloud.google.com/tpu/docs/run-in-container#train_a_jax_model_in_a_docker_container # follow the instructions here: https://cloud.google.com/tpu/docs/run-in-container#train_a_jax_model_in_a_docker_container
RUN python3 -m pip install --no-cache-dir --upgrade pip uv && \ RUN python3 -m pip install --no-cache-dir --upgrade pip uv && \
python3 -m uv pip install --no-cache-dir \ python3 -m pip install --no-cache-dir \
"jax[tpu]>=0.2.16,!=0.3.2" \ "jax[tpu]>=0.2.16,!=0.3.2" \
-f https://storage.googleapis.com/jax-releases/libtpu_releases.html && \ -f https://storage.googleapis.com/jax-releases/libtpu_releases.html && \
python3 -m uv pip install --upgrade --no-cache-dir \ python3 -m uv pip install --upgrade --no-cache-dir \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment