checkpoint/fsdp_config.json

{
    "FSDP_version": 1,
    "world_size": 4
}