Skip to content

Commit cdaedde

Browse files
committed
update sh scripts
1 parent f4e11b0 commit cdaedde

File tree

2 files changed

+4
-14
lines changed

2 files changed

+4
-14
lines changed
Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,4 @@
1-
#!/usr/bin/env bash
2-
31
MEGATRON_LM_PATH=${MEGATRON_LM_PATH:-/path/to/Megatron-LM}
4-
ASCEND_RT_VISIBLE_DEVICES=${ASCEND_RT_VISIBLE_DEVICES:-0,1,2,3,4,5,6,7}
5-
6-
export PYTHONPATH="${MEGATRON_LM_PATH}:${PYTHONPATH:-}"
7-
8-
ASCEND_RT_VISIBLE_DEVICES="${ASCEND_RT_VISIBLE_DEVICES}" \
2+
ASCEND_RT_VISIBLE_DEVICES=${ASCEND_RT_VISIBLE_DEVICES:-0,1,2,3,4,5,6,7} \
3+
PYTHONPATH="${MEGATRON_LM_PATH}:${PYTHONPATH:-}" \
94
torchrun --nproc_per_node=8 cookbook/megatron/npu/tp_lora_npu.py
Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,4 @@
1-
#!/usr/bin/env bash
2-
31
MEGATRON_LM_PATH=${MEGATRON_LM_PATH:-/path/to/Megatron-LM}
4-
ASCEND_RT_VISIBLE_DEVICES=${ASCEND_RT_VISIBLE_DEVICES:-0,1,2,3,4,5,6,7}
5-
6-
export PYTHONPATH="${MEGATRON_LM_PATH}:${PYTHONPATH:-}"
7-
8-
ASCEND_RT_VISIBLE_DEVICES="${ASCEND_RT_VISIBLE_DEVICES}" \
2+
ASCEND_RT_VISIBLE_DEVICES=${ASCEND_RT_VISIBLE_DEVICES:-0,1,2,3,4,5,6,7} \
3+
PYTHONPATH="${MEGATRON_LM_PATH}:${PYTHONPATH:-}" \
94
torchrun --nproc_per_node=8 cookbook/megatron/npu/tp_moe_lora_npu.py

0 commit comments

Comments
 (0)