style: apply ruff formatting to FSDP2 changes

This commit is contained in:
2026-07-23 16:30:10 +08:00
parent 4e38183e86
commit 021e6f3788
2 changed files with 4 additions and 4 deletions
+3 -1
View File
@@ -472,7 +472,9 @@ def train(
]
assert os.path.exists(param_path)
if nprocs > 1 and parallel_mode == "none":
raise ValueError("--nprocs > 1 requires --parallel_mode to be 'ddp', 'fsdp', or 'fsdp2'")
raise ValueError(
"--nprocs > 1 requires --parallel_mode to be 'ddp', 'fsdp', or 'fsdp2'"
)
# Load config
config_path = os.path.join(param_path, "config.json")