Skip to content

Commit 911c082

Browse files
committed
Remove Megatron shared shim
1 parent 0cee7cf commit 911c082

2 files changed

Lines changed: 19 additions & 20 deletions

File tree

src/art/megatron/shared.py

Lines changed: 0 additions & 20 deletions
This file was deleted.

src/art/megatron/train.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,14 @@
44
configure_megatron_runtime_env()
55
# isort: on
66

7+
"""Megatron training runtime and public worker API.
8+
9+
Public cross-repo API consumed by serverless-training:
10+
- build_training_runtime
11+
- run_megatron_worker_loop
12+
- merge_lora_adapter
13+
"""
14+
715
import gc
816
import importlib
917
import json
@@ -58,6 +66,17 @@
5866

5967
DEFAULT_MODEL_IDENTIFIER = "Qwen/Qwen3-30B-A3B-Instruct-2507"
6068

69+
__all__ = [
70+
"DEFAULT_MODEL_IDENTIFIER",
71+
"TrainingRuntime",
72+
"build_training_runtime",
73+
"run_megatron_worker_loop",
74+
"run_megatron_rl_job",
75+
"run_megatron_sft_job",
76+
"finalize_megatron_job",
77+
"merge_lora_adapter",
78+
]
79+
6180

6281
class TrainingRuntime(BaseModel):
6382
model_config = ConfigDict(arbitrary_types_allowed=True)

0 commit comments

Comments
 (0)