Update on "[ET-VK][ez] Accept sample_kwargs as an argument in several test util functions"

Title says it all! This makes it possible to export models that require kwargs to be defined instead of args.

Differential Revision: [D84716455](https://our.internmc.facebook.com/intern/diff/D84716455/)

[ghstack-poisoned]

Author: ssjia