fix binding to the wrong python module (#67246)
Summary:
Fixes #{issue number}
Pull Request resolved: https://github.com/pytorch/pytorch/pull/67246
Reviewed By: zhxchen17
Differential Revision: D31937065
Pulled By: Krovatkin
fbshipit-source-id: 5c10b2870bccece50ba52dde26127da79bccbba6
