chore: install optimum-tpu from pypi instead of git

huggingface · Jan 15, 2025 · 505ad5a · 505ad5a
1 parent 0a4dcbd
commit 505ad5a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/containers/pytorch/training/tpu/2.5.1/transformers/4.46.3/py310/Dockerfile b/containers/pytorch/training/tpu/2.5.1/transformers/4.46.3/py310/Dockerfile
@@ -30,7 +30,7 @@ RUN pip install --upgrade --no-cache-dir \
   jupyter notebook
 
 # Install Optimum TPU
-RUN pip install git+https://github.com/huggingface/optimum-tpu.git@v${OPTIMUM_TPU}
+RUN pip install --upgrade --no-cache-dir optimum-tpu==${OPTIMUM_TPU}
 
 # Install Google CLI single command
 RUN echo "deb [signed-by=/usr/share/keyrings/cloud.google.gpg] https://packages.cloud.google.com/apt cloud-sdk main" \