mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Revert "Grouped Query Attention (#128898)"
This reverts commit d039b14207fe659d664c590efc06cc0a2abc96c0. Reverted https://github.com/pytorch/pytorch/pull/128898 on behalf of https://github.com/albanD due to Broken test on main ([comment](https://github.com/pytorch/pytorch/pull/128898#issuecomment-2258314481))
This commit is contained in:
@ -466,7 +466,6 @@ def gen_nn_functional(fm: FileManager) -> None:
|
||||
"dropout_p: float = 0.0",
|
||||
"is_causal: bool = False",
|
||||
"scale: Optional[float] = None",
|
||||
"enable_gqa: bool = False",
|
||||
]
|
||||
)
|
||||
)
|
||||
|
Reference in New Issue
Block a user