Skip to content

Commit c18315b

Browse files
Fix resuming
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
1 parent ce4d081 commit c18315b

1 file changed

Lines changed: 0 additions & 2 deletions

File tree

examples/megatron_bridge/distill.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,6 @@
7272
max_sequence_length=256000,
7373
)
7474
"""
75-
# TODO: Fix resuming distillation from an intermediate checkpoint.
7675

7776
import argparse
7877
import os
@@ -278,7 +277,6 @@ def _build_model_provider(hf_path):
278277
ckpt_format="torch_dist",
279278
async_save=True,
280279
fully_parallel_save=True,
281-
finetune=True,
282280
),
283281
rng=RNGConfig(seed=SEED),
284282
mixed_precision="bf16_mixed",

0 commit comments

Comments
 (0)