migrate example configs to canonical attn_implementation

2026-04-23 22:15:07 +00:00
parent 2d64d009d8
commit 39226623d2
222 changed files with 209 additions and 243 deletions
--- a/examples/llama-4/scout-qlora-flexattn-fsdp2.yaml
+++ b/examples/llama-4/scout-qlora-flexattn-fsdp2.yaml
@@ -59,7 +59,7 @@ bf16: true
 tf32: true

 logging_steps: 1
-flex_attention: true
+attn_implementation: flex_attention
 flex_attn_compile_kwargs:
  dynamic: false
  mode: max-autotune-no-cudagraphs