minor

kevalmorabia97 · kevalmorabia97 · commit da2599ccb099 · 2026-04-13T10:55:52.000-07:00
Signed-off-by: Keval Morabia &lt;28916987+kevalmorabia97@users.noreply.github.com&gt;
diff --git a/examples/megatron_bridge/distill.py b/examples/megatron_bridge/distill.py
@@ -332,6 +332,7 @@ def _build_model_provider(hf_path):
             export_bridge = AutoBridge.from_hf_pretrained(
                 args.student_hf_model, trust_remote_code=args.trust_remote_code
             )
+            # Copy weights and remote code
             export_bridge.export_ckpt(
                 megatron_path=f"{checkpoint_dir}/iter_{args.train_iters:07d}",
                 hf_path=args.hf_export_path,

Original file line number	Diff line number	Diff line change
`@@ -332,6 +332,7 @@ def _build_model_provider(hf_path):`
`332`	`332`	`export_bridge = AutoBridge.from_hf_pretrained(`
`333`	`333`	`args.student_hf_model, trust_remote_code=args.trust_remote_code`
`334`	`334`	`)`
	`335`	`+ # Copy weights and remote code`
`335`	`336`	`export_bridge.export_ckpt(`
`336`	`337`	`megatron_path=f"{checkpoint_dir}/iter_{args.train_iters:07d}",`
`337`	`338`	`hf_path=args.hf_export_path,`