From 3e3d46924b4fd1fa36ccbc5605ed619cdcc00dbf Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Tue, 27 Feb 2024 16:25:50 +0530 Subject: [PATCH] [Dockerfile] remove uv from docker jax tpu (#7115) remove uv from docker jax tpu --- docker/diffusers-flax-tpu/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/diffusers-flax-tpu/Dockerfile b/docker/diffusers-flax-tpu/Dockerfile index 4b38dd13e4..c0722139ca 100644 --- a/docker/diffusers-flax-tpu/Dockerfile +++ b/docker/diffusers-flax-tpu/Dockerfile @@ -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) # 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 && \ - python3 -m uv pip install --no-cache-dir \ + python3 -m pip install --no-cache-dir \ "jax[tpu]>=0.2.16,!=0.3.2" \ -f https://storage.googleapis.com/jax-releases/libtpu_releases.html && \ python3 -m uv pip install --upgrade --no-cache-dir \