[AMD] Use bridge branch for ROCm Megatron-Bridge (#1475)

Co-authored-by: Zhiyao Jiang <jessicajiang324@gmail.com>
This commit is contained in:
Xinyu Jiang
2026-06-25 21:41:04 -07:00
committed by GitHub
co-authored by Zhiyao Jiang
parent e80610bfd0
commit 39a1580e2c
+1 -1
View File
@@ -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