fix

2025-06-01 20:05:27 +00:00 · 2024-11-19 11:55:02 +08:00 · 2024-11-19 11:55:02 +08:00 · 603d06ad56
commit 603d06ad56
parent 945a67dd61
1 changed files with 1 additions and 1 deletions
--- a/tests/test_checkpoint_io/test_gemini_checkpoint_io.py
+++ b/tests/test_checkpoint_io/test_gemini_checkpoint_io.py
@ -192,7 +192,7 @@ def run_dist(rank, world_size, port):
    colossalai.launch(rank=rank, world_size=world_size, host="localhost", port=port, backend="nccl")
    exam_state_dict()
    exam_state_dict_with_origin()
-    # exam_lazy_from_pretrained()
+    exam_lazy_from_pretrained()


@pytest.mark.dist