From f18925fb4bb7af8f4a6e690242e24cef6249e2c0 Mon Sep 17 00:00:00 2001
From: NanoCode012 <kevinvong@rocketmail.com>
Date: Wed, 14 Aug 2024 22:46:46 +0900
Subject: [PATCH] fix: parse eager_attention (#1824)

---
 src/axolotl/utils/config/models/input/v0_4_1/__init__.py | 2 ++
 1 file changed, 2 insertions(+)

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
index b765263ba6..7bf77965c2 100644
--- 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