更新0.12

160bf237 · wangxj · b01809dd · 160bf237 · 160bf237 · 160bf237
Commit 160bf237 authored Mar 05, 2025 by wangxj
20 changed files
--- a/megatron/core/inference/scheduler.py
+++ b/megatron/core/inference/scheduler.py
--- a/megatron/core/inference/text_generation_controllers/__init__.py
+++ b/megatron/core/inference/text_generation_controllers/__init__.py
--- a/megatron/core/inference/text_generation_controllers/encoder_decoder_text_generation_controller.py
+++ b/megatron/core/inference/text_generation_controllers/encoder_decoder_text_generation_controller.py
--- a/megatron/core/inference/text_generation_controllers/simple_text_generation_controller.py
+++ b/megatron/core/inference/text_generation_controllers/simple_text_generation_controller.py
--- a/megatron/core/inference/text_generation_controllers/text_generation_controller.py
+++ b/megatron/core/inference/text_generation_controllers/text_generation_controller.py
--- a/megatron/core/inference/text_generation_controllers/vlm_text_generation_controller.py
+++ b/megatron/core/inference/text_generation_controllers/vlm_text_generation_controller.py
--- a/megatron/core/inference/utils.py
+++ b/megatron/core/inference/utils.py
--- a/megatron/core/inference_params.py
+++ b/megatron/core/inference_params.py
--- a/megatron/core/jit.py
+++ b/megatron/core/jit.py
--- a/megatron/core/model_parallel_config.py
+++ b/megatron/core/model_parallel_config.py
--- a/megatron/core/models/T5/__init__.py
+++ b/megatron/core/models/T5/__init__.py
--- a/megatron/core/models/T5/t5_model.py
+++ b/megatron/core/models/T5/t5_model.py
--- a/megatron/core/models/T5/t5_spec.py
+++ b/megatron/core/models/T5/t5_spec.py
--- a/megatron/core/models/__init__.py
+++ b/megatron/core/models/__init__.py
--- a/megatron/core/models/bert/__init__.py
+++ b/megatron/core/models/bert/__init__.py
--- a/megatron/core/models/bert/bert_layer_specs.py
+++ b/megatron/core/models/bert/bert_layer_specs.py
--- a/megatron/core/models/bert/bert_lm_head.py
+++ b/megatron/core/models/bert/bert_lm_head.py
--- a/megatron/core/models/bert/bert_model.py
+++ b/megatron/core/models/bert/bert_model.py
--- a/megatron/core/models/bert/pooler.py
+++ b/megatron/core/models/bert/pooler.py
--- a/megatron/core/models/common/__init__.py
+++ b/megatron/core/models/common/__init__.py