FSDP2 with offload for SFT

[agenticgogogo]

Hi authors, thank you so much for the contribution.

For fsdp2, should cpu_offload be CPUOffloadPolicy, instead of CPUOffload? It now reports "RuntimeError: attempting to assign a gradient with device type 'cuda' to a tensor with device type 'cpu'"