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/swanlab/lora-swanlab-profiling.yml
+++ b/examples/swanlab/lora-swanlab-profiling.yml
@@ -59,7 +59,7 @@ tf32: false

 # Performance
 gradient_checkpointing: true
-flash_attention: true
+attn_implementation: flash_attention_2

 # Checkpointing and Logging
 logging_steps: 1