Skip to content

Commit 501b5d5

Browse files
committed
Update MXFP8 refit test for source metadata API
Signed-off-by: seonjinn <sna@nvidia.com>
1 parent ecb9aa0 commit 501b5d5

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

‎tests/unit/models/policy/test_megatron_worker.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2745,7 +2745,7 @@ def _iter_params():
27452745

27462746
worker._iter_params_with_optional_kv_scales = _iter_params
27472747

2748-
info = worker._prepare_source_refit_info("hf_export")
2748+
info = worker._prepare_source_refit_info()
27492749

27502750
assert worker._refit_prequant_names == set()
27512751
assert info[name] == (torch.Size([2, 64]), torch.bfloat16)

0 commit comments

Comments
 (0)