fix: parse eager_attention (#1824)

axolotl-ai-cloud · Aug 14, 2024 · f18925f · f18925f
1 parent 1853d60
commit f18925f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py
@@ -614,6 +614,8 @@ class Config:
     flash_attn_fuse_mlp: Optional[bool] = None
     flash_optimum: Optional[bool] = None
 
+    eager_attention: Optional[bool] = None
+
     unsloth_cross_entropy_loss: Optional[bool] = None
     unsloth_lora_mlp: Optional[bool] = None
     unsloth_lora_qkv: Optional[bool] = None