Skip to content

Commit b30a725

Browse files
committed
clean up
Signed-off-by: Li Ding <liding@nvidia.com>
1 parent 8fc0f7d commit b30a725

3 files changed

Lines changed: 0 additions & 9 deletions

File tree

‎src/megatron/bridge/models/nemotronh/__init__.py‎

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -13,14 +13,8 @@
1313
# limitations under the License.
1414

1515
from megatron.bridge.models.nemotronh.nemotron_h_bridge import NemotronHBridge
16-
from megatron.bridge.models.nemotronh.nemotron_h_provider import (
17-
Nemotron3SuperDebugProvider,
18-
Nemotron3SuperProvider,
19-
)
2016

2117

2218
__all__ = [
2319
"NemotronHBridge",
24-
"Nemotron3SuperProvider",
25-
"Nemotron3SuperDebugProvider",
2620
]

‎src/megatron/bridge/recipes/nemotronh/__init__.py‎

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@
2222

2323
# Nemotron 3 Super models
2424
from megatron.bridge.recipes.nemotronh.nemotron_3_super import (
25-
nemotron_3_super_debug_pretrain_config,
2625
nemotron_3_super_finetune_config,
2726
nemotron_3_super_pretrain_config,
2827
)
@@ -79,6 +78,5 @@
7978
"nemotron_3_nano_peft_config",
8079
# Nemotron 3 Super models
8180
"nemotron_3_super_pretrain_config",
82-
"nemotron_3_super_debug_pretrain_config",
8381
"nemotron_3_super_finetune_config",
8482
]

‎src/megatron/bridge/recipes/nemotronh/nemotron_3_super.py‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -261,6 +261,5 @@ def nemotron_3_super_finetune_config(
261261

262262
__all__ = [
263263
"nemotron_3_super_pretrain_config",
264-
"nemotron_3_super_debug_pretrain_config",
265264
"nemotron_3_super_finetune_config",
266265
]

0 commit comments

Comments
 (0)