[FSDP][Example] Fix FSDP example for use_orig_params=False
#5312
| Job | Run time |
|---|---|
| 1m 49s | |
| 2m 55s | |
| 2m 42s | |
| 2m 30s | |
| 2m 27s | |
| 2m 19s | |
| 2m 19s | |
| 4m 0s | |
| 3m 38s | |
| 8m 31s | |
| 2m 44s | |
| 4m 4s | |
| 3m 40s | |
| 3m 21s | |
| 3m 21s | |
| 3m 34s | |
| 3m 17s | |
| 4m 56s | |
| 4m 46s | |
| 11m 25s | |
| 1h 18m 18s |