migrate example configs to canonical attn_implementation
This commit is contained in:
@@ -78,7 +78,7 @@ tf32: false
|
||||
|
||||
# Performance
|
||||
gradient_checkpointing: true
|
||||
flash_attention: true
|
||||
attn_implementation: flash_attention_2
|
||||
|
||||
# Checkpointing and Logging
|
||||
logging_steps: 1
|
||||
|
||||
@@ -102,7 +102,7 @@ bf16: auto
|
||||
tf32: false
|
||||
|
||||
gradient_checkpointing: true
|
||||
flash_attention: true
|
||||
attn_implementation: flash_attention_2
|
||||
|
||||
# ============================================================================
|
||||
# Checkpointing and Logging
|
||||
|
||||
@@ -59,7 +59,7 @@ tf32: false
|
||||
|
||||
# Performance
|
||||
gradient_checkpointing: true
|
||||
flash_attention: true
|
||||
attn_implementation: flash_attention_2
|
||||
|
||||
# Checkpointing and Logging
|
||||
logging_steps: 1
|
||||
|
||||
Reference in New Issue
Block a user