From e7eed203d866f17fd92d5434ca161a2f908fe236 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Mon, 26 Feb 2024 14:36:37 -0500
Subject: [PATCH] hotfix for missing outputs params (#1333)

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

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 44111b577..f0ab7f235 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
@@ -336,6 +336,7 @@ class WandbConfig(BaseModel):
 # pylint: disable=too-many-public-methods,too-many-ancestors
 class AxolotlInputConfig(
     ModelInputConfig,
+    ModelOutputConfig,
     LoraConfig,
     ReLoRAConfig,
     HyperparametersConfig,