mirror of
https://github.com/radixark/miles.git
synced 2026-10-02 07:14:53 +08:00
[AMD] Use bridge branch for ROCm Megatron-Bridge (#1475)
Co-authored-by: Zhiyao Jiang <jessicajiang324@gmail.com>
This commit is contained in:
co-authored by
Zhiyao Jiang
parent
e80610bfd0
commit
39a1580e2c
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user