fixing call_module on subscripting into generator (#81258)
named_modules() return a generator, which is not subscriptable and causes node support query to fail
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81258
Approved by: https://github.com/SherlockNoMad