migrate example configs to canonical attn_implementation

This commit is contained in:
Wing Lian
2026-04-23 22:15:07 +00:00
parent 2d64d009d8
commit 39226623d2
222 changed files with 209 additions and 243 deletions

View File

@@ -78,7 +78,7 @@ tf32: false
# Performance
gradient_checkpointing: true
flash_attention: true
attn_implementation: flash_attention_2
# Checkpointing and Logging
logging_steps: 1

View File

@@ -102,7 +102,7 @@ bf16: auto
tf32: false
gradient_checkpointing: true
flash_attention: true
attn_implementation: flash_attention_2
# ============================================================================
# Checkpointing and Logging

View File

@@ -59,7 +59,7 @@ tf32: false
# Performance
gradient_checkpointing: true
flash_attention: true
attn_implementation: flash_attention_2
# Checkpointing and Logging
logging_steps: 1