Fix fx tests with inputs_embeds (#31862)

Commit

1 year ago

Fix fx tests with inputs_embeds (#31862) * fix tests * [test_all] check * address review comments