Skip to content

Commit 871b5d7

Browse files
bvandermoonGoogle-ML-Automation
authored andcommitted
Fix exclusion path for maxtext_tpu_dependencies.Dockerfile.
PiperOrigin-RevId: 882212987
1 parent 8e0aaf5 commit 871b5d7

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/dependencies/dockerfiles/maxtext_tpu_dependencies.Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ ENV ENV_DEVICE=$DEVICE
3737

3838
ENV MAXTEXT_ASSETS_ROOT=/deps/src/maxtext/assets
3939
ENV MAXTEXT_TEST_ASSETS_ROOT=/deps/tests/assets
40-
ENV MAXTEXT_PKG_DIR=/deps/src/MaxText
40+
ENV MAXTEXT_PKG_DIR=/deps/src/maxtext
4141
ENV MAXTEXT_REPO_ROOT=/deps
4242

4343
# Set the working directory in the container
@@ -47,7 +47,7 @@ WORKDIR /deps
4747
COPY tools/setup tools/setup/
4848
COPY src/dependencies/requirements/ src/dependencies/requirements/
4949
COPY src/install_maxtext_extra_deps/ src/install_maxtext_extra_deps/
50-
COPY src/MaxText/integration/vllm/ src/MaxText/integration/vllm/
50+
COPY src/maxtext/integration/vllm/ src/maxtext/integration/vllm/
5151

5252
# Copy the custom libtpu.so file if it exists inside maxtext repository
5353
COPY libtpu.so* /root/custom_libtpu/

0 commit comments

Comments
 (0)