realign/train/configs/fsdp/qwen_fsdp_config.json

{
    "fsdp_transformer_layer_cls_to_wrap": ["QWenBlock"],
    "limit_all_gathers": true
}