We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2790ed2 commit 3d6a5deCopy full SHA for 3d6a5de
1 file changed
dependencies/dockerfiles/maxtext_tpu_dependencies.Dockerfile
@@ -37,7 +37,7 @@ ENV ENV_DEVICE=$DEVICE
37
38
ENV MAXTEXT_ASSETS_ROOT=/deps/src/maxtext/assets
39
ENV MAXTEXT_TEST_ASSETS_ROOT=/deps/tests/assets
40
-ENV MAXTEXT_PKG_DIR=/deps/src/MaxText
+ENV MAXTEXT_PKG_DIR=/deps/src/maxtext
41
ENV MAXTEXT_REPO_ROOT=/deps
42
43
# Set the working directory in the container
@@ -47,7 +47,7 @@ WORKDIR /deps
47
COPY tools/setup tools/setup/
48
COPY dependencies/requirements/ dependencies/requirements/
49
COPY src/install_maxtext_extra_deps/ src/install_maxtext_extra_deps/
50
-COPY src/MaxText/integration/vllm/ src/MaxText/integration/vllm/
+COPY src/maxtext/integration/vllm/ src/maxtext/integration/vllm/
51
52
# Copy the custom libtpu.so file if it exists inside maxtext repository
53
COPY libtpu.so* /root/custom_libtpu/
0 commit comments