We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1a84728 commit 089234cCopy full SHA for 089234c
swift/llm/megatron/utils.py
@@ -29,6 +29,8 @@ def init_megatron_env() -> None:
29
megatron_patch_path = git_clone_github(
30
'https://github.com/alibaba/Pai-Megatron-Patch', commit_hash='6fd5d050b240fd959f0ba69f1e9cd9a053e5a81d')
31
os.environ['PAI_MEGATRON_PATCH_PATH'] = megatron_patch_path
32
+ else:
33
+ megatron_patch_path = os.environ['PAI_MEGATRON_PATCH_PATH']
34
sys.path.append(os.environ['PAI_MEGATRON_PATCH_PATH'])
35
36
# rename qwen1.5->qwen1_5 files
0 commit comments