From a86cea5708d3a64ad95e621a843e46640bffb4e6 Mon Sep 17 00:00:00 2001 From: Pepijn Date: Tue, 14 Oct 2025 11:26:57 +0200 Subject: [PATCH] fix path optimizer state --- tests/training/test_multi_gpu.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/training/test_multi_gpu.py b/tests/training/test_multi_gpu.py index c24af6138..3b9cc8834 100644 --- a/tests/training/test_multi_gpu.py +++ b/tests/training/test_multi_gpu.py @@ -218,5 +218,5 @@ class TestMultiGPUTraining: assert training_state_dir.exists(), f"No training state in checkpoint {checkpoint_dir}" # Verify optimizer state exists - optimizer_state = training_state_dir / "optimizer_state.pt" + optimizer_state = training_state_dir / "optimizer_state.safetensors" assert optimizer_state.exists(), f"No optimizer state in checkpoint {checkpoint_dir}"