tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

replace attention in the yaml config with an enum

Browse Source
This commit is contained in:
Wing Lian
2025-04-04 23:37:30 -04:00
committed by NanoCode012
parent 0d71b0aa5f
commit ba47adc24b
101 changed files with 268 additions and 122 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/pixtral/lora-12b.yml
View File

@@ -46,8 +46,7 @@ tf32: true
gradient_checkpointing: true
logging_steps: 1
flash_attention: false # PixtralVisionModel does not support Flash Attention 2.0 yet
eager_attention:
attention: eager # PixtralVisionModel does not support Flash Attention 2.0 yet
warmup_ratio: 0.1
evals_per_epoch: 1