Fix old annotate() error (#25261)
Summary:
Fixes #25067
](https://our.intern.facebook.com/intern/diff/17103889/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/25261
Pulled By: driazati
Differential Revision: D17103889
fbshipit-source-id: bd94cb36cf4829e63ad39ae169047b9b9e857679