Update saver_legacy.py

a65d5678 · wxj · 63c300ba · a65d5678
Commit a65d5678 authored Apr 17, 2025 by wxj
Show whitespace changes
Inline Side-by-side

Showing with 415 additions and 415 deletions

tools/checkpoint/saver_legacy.py tools/checkpoint/saver_legacy.py +415 -415

No files found.
--- a/tools/checkpoint/saver_legacy.py
+++ b/tools/checkpoint/saver_legacy.py
@@ -206,7 +206,7 @@ def save_checkpoint(queue, args):
    mpu.set_pipeline_model_parallel_world_size(args.target_pipeline_parallel_size)
    mpu.set_tensor_model_parallel_rank(0)
    mpu.set_pipeline_model_parallel_rank(0)
-    fused_kernels.load(margs)
+    # fused_kernels.load(margs)

    # Embeddings
    # -----------