From 39a1580e2c8f48d8ec01c6a29e0d07051c4fb3b9 Mon Sep 17 00:00:00 2001 From: Xinyu Jiang Date: Fri, 26 Jun 2026 00:41:04 -0400 Subject: [PATCH] [AMD] Use bridge branch for ROCm Megatron-Bridge (#1475) Co-authored-by: Zhiyao Jiang --- docker/Dockerfile.rocm | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile.rocm b/docker/Dockerfile.rocm index 3e62424d0a..6e90443543 100644 --- a/docker/Dockerfile.rocm +++ b/docker/Dockerfile.rocm @@ -120,7 +120,7 @@ RUN cd /sgl-workspace/sglang && \ RUN python -c "import sglang; import sgl_kernel; print('SGLang + sgl_kernel: OK')" RUN pip install git+https://github.com/fzyzcjy/torch_memory_saver.git@d64a639 --no-cache-dir --force-reinstall -RUN pip install git+https://github.com/radixark/Megatron-Bridge.git@main --no-deps --no-build-isolation +RUN pip install git+https://github.com/radixark/Megatron-Bridge.git@bridge --no-deps --no-build-isolation RUN pip install megatron-energon --no-deps RUN pip install multi-storage-client --no-deps