[Quant] Fixed errors in test_embedding introduced by https://github.com/pytorch/pytorch/pull/69768 (#71387)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/71387
Test Plan: Imported from OSS
Reviewed By: albanD
Differential Revision: D33808593
Pulled By: dzdang
fbshipit-source-id: 3950400dc7506006666fcd055819e9a08a42eda9
(cherry picked from commit 38dc2de49d0f53b0bc650e70ef98410b6432face)
