Fix (example/llm): Removed unnecessary intantiation kwarg

Xilinx · Nov 28, 2024 · 01ef9e3 · 01ef9e3
1 parent 32c94b9
commit 01ef9e3
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/brevitas_examples/llm/main.py b/src/brevitas_examples/llm/main.py
@@ -187,8 +187,6 @@ def main(args):
     kwargs = {"torch_dtype": dtype}
     if quant_sdpa_fx:
         kwargs["attn_implementation"] = "sdpa"
-    elif args.replace_mha:
-        kwargs["attn_implementation"] = "eager"
 
     if args.export_target == 'torch_qcdq':
         kwargs['torchscript'] = True