Skip to content

Commit 089234c

Browse files
authored
fix megatron_patch_path (#1804)
1 parent 1a84728 commit 089234c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

swift/llm/megatron/utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,8 @@ def init_megatron_env() -> None:
2929
megatron_patch_path = git_clone_github(
3030
'https://github.com/alibaba/Pai-Megatron-Patch', commit_hash='6fd5d050b240fd959f0ba69f1e9cd9a053e5a81d')
3131
os.environ['PAI_MEGATRON_PATCH_PATH'] = megatron_patch_path
32+
else:
33+
megatron_patch_path = os.environ['PAI_MEGATRON_PATCH_PATH']
3234
sys.path.append(os.environ['PAI_MEGATRON_PATCH_PATH'])
3335

3436
# rename qwen1.5->qwen1_5 files

0 commit comments

Comments
 (0)