From 7db6de44c763796ae2dc93ad8138f5cfa220270d Mon Sep 17 00:00:00 2001 From: Nick Fraser Date: Fri, 29 Nov 2024 10:09:25 +0000 Subject: [PATCH] test (nn/sdpa): bugfix --- src/brevitas/nn/quant_sdpa.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/brevitas/nn/quant_sdpa.py b/src/brevitas/nn/quant_sdpa.py index e9e6d89c9..96e85d489 100644 --- a/src/brevitas/nn/quant_sdpa.py +++ b/src/brevitas/nn/quant_sdpa.py @@ -100,7 +100,7 @@ def forward( kwargs = {} if scale is not None: kwargs["scale"] = scale - if not enable_gqa: + if enable_gqa: kwargs["enable_gqa"] = enable_gqa return F.scaled_dot_product_attention( query=query,