File tree Expand file tree Collapse file tree
src/dependencies/dockerfiles Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -37,7 +37,7 @@ ENV ENV_DEVICE=$DEVICE
3737
3838ENV MAXTEXT_ASSETS_ROOT=/deps/src/maxtext/assets
3939ENV MAXTEXT_TEST_ASSETS_ROOT=/deps/tests/assets
40- ENV MAXTEXT_PKG_DIR=/deps/src/MaxText
40+ ENV MAXTEXT_PKG_DIR=/deps/src/maxtext
4141ENV MAXTEXT_REPO_ROOT=/deps
4242
4343# Set the working directory in the container
@@ -47,7 +47,7 @@ WORKDIR /deps
4747COPY tools/setup tools/setup/
4848COPY src/dependencies/requirements/ src/dependencies/requirements/
4949COPY 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
5353COPY libtpu.so* /root/custom_libtpu/
You can’t perform that action at this time.
0 commit comments