Fix `BertGeneration` (#35043)

Commit

1 year ago

Fix `BertGeneration` (#35043) fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>