diff --git a/vllm/Dockerfile b/vllm/Dockerfile index 3030348..5564f10 100644 --- a/vllm/Dockerfile +++ b/vllm/Dockerfile @@ -166,7 +166,9 @@ RUN cd vllm && \ echo "========================================\n\n" && \ git submodule sync && \ git submodule update --init --recursive -j 8 && \ - sed -i 's/GIT_TAG [a-f0-9]\{40\}/GIT_TAG main/' cmake/external_projects/vllm_flash_attn.cmake && \ + # NOTE: Removed the sed that forced flash-attn GIT_TAG to main. + # v0.19.0 pins a compatible commit; building from main causes + # FLASHATTENTION_FP8_TWO_LEVEL_INTERVAL undefined errors. sed -i 's/register_opaque_type(ModuleName, typ="value", hoist=True)/register_opaque_type(ModuleName, typ="value")/' vllm/utils/torch_utils.py && \ export MAX_JOBS=8 && \ export CMAKE_BUILD_PARALLEL_LEVEL=$MAX_JOBS && \