修改mtp

770fa304 · dongcl · 8096abd4 · 770fa304 · 770fa304 · 770fa304
Commit 770fa304 authored Apr 25, 2025 by dongcl
4 changed files
--- a/dcu_megatron/core/utils.py
+++ b/dcu_megatron/core/utils.py
--- a/dcu_megatron/training/arguments.py
+++ b/dcu_megatron/training/arguments.py
--- a/dcu_megatron/training/utils.py
+++ b/dcu_megatron/training/utils.py
--- a/pretrain_gpt.py
+++ b/pretrain_gpt.py