Fix bug where quantization framework does not work with training

[ryanbogan]

Description

Currently, the quantization framework is broken and does not work when training. This PR fixes this bug, so the vectors are properly quantized when the framework is enabled

Check List

• [x] New functionality includes testing.
• [x] New functionality has been documented.
• [x] API changes companion pull request created.
• [x] Commits are signed per the DCO using --signoff.
• [x] Public documentation issue/PR created.

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license. For more information on following Developer Certificate of Origin and signing off your commits, please check here.

[ryanbogan]

Rolling upgrade failure not related to this PR