TrainerIntegrationFSDP::test_basic_run_with_cpu_offload
torch < 2.1
* fix * fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>