add

0d99ae1f · silencealiang · c271aaae · 0d99ae1f · 0d99ae1f · 0d99ae1f
Commit 0d99ae1f authored Mar 14, 2025 by silencealiang
20 changed files
--- a/megatron/training/arguments.py
+++ b/megatron/training/arguments.py
--- a/megatron/training/async_utils.py
+++ b/megatron/training/async_utils.py
--- a/megatron/training/checkpointing.py
+++ b/megatron/training/checkpointing.py
--- a/megatron/training/dist_signal_handler.py
+++ b/megatron/training/dist_signal_handler.py
--- a/megatron/training/ft_integration.py
+++ b/megatron/training/ft_integration.py
--- a/megatron/training/global_vars.py
+++ b/megatron/training/global_vars.py
--- a/megatron/training/initialize.py
+++ b/megatron/training/initialize.py
--- a/megatron/training/log_handler.py
+++ b/megatron/training/log_handler.py
--- a/megatron/training/one_logger_utils.py
+++ b/megatron/training/one_logger_utils.py
--- a/megatron/training/theoretical_memory_usage.py
+++ b/megatron/training/theoretical_memory_usage.py
--- a/megatron/training/tokenizer/__init__.py
+++ b/megatron/training/tokenizer/__init__.py
--- a/megatron/training/tokenizer/bert_tokenization.py
+++ b/megatron/training/tokenizer/bert_tokenization.py
--- a/megatron/training/tokenizer/gpt2_tokenization.py
+++ b/megatron/training/tokenizer/gpt2_tokenization.py
--- a/megatron/training/tokenizer/multimodal_tokenizer.py
+++ b/megatron/training/tokenizer/multimodal_tokenizer.py
--- a/megatron/training/tokenizer/tokenizer.py
+++ b/megatron/training/tokenizer/tokenizer.py
--- a/megatron/training/training.py
+++ b/megatron/training/training.py
--- a/megatron/training/utils.py
+++ b/megatron/training/utils.py
--- a/megatron/training/yaml_arguments.py
+++ b/megatron/training/yaml_arguments.py
--- a/mypy.ini
+++ b/mypy.ini
--- a/pretrain_bert.py
+++ b/pretrain_bert.py