pytorch

3ac27e78 - Fix typehint of multi_head_attention_forward

Fix typehint of multi_head_attention_forward Fixes #76169 Pull Request resolved: https://github.com/pytorch/pytorch/pull/76170 Approved by: https://github.com/jbschlosser