generate with dtensor policy

[gshennvm]

dtensor policy should support generate

[github-actions]

This issue is stale because it has been open for 30 days with no activity. Remove stale label or comment or this will be closed in 7 days.

[SahilJain314]

Since we've deprecated HF generate across the board, I'm happy to close this.