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-3/3b-fp8-fsdp2.yaml
+++ b/examples/llama-3/3b-fp8-fsdp2.yaml
@@ -25,7 +25,7 @@ sample_packing: true
 pad_to_sequence_len: true
 sequence_len: 512

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