Allow specifying the number of quantization groups in the inference test script using a quantize_groups argument.
This PR is complimentary to PR #3519 on the main repo, and should be merged with (or after) merging PR #3519.
Tagging @RezaYazdaniAminabadi as a reviewer on #3519.