From 60763b2e61bc0a9fc3110f978d8aace9547af79f Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Thu, 14 Nov 2024 10:14:13 -0500 Subject: [PATCH] fix missing return --- 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 10899d1f9..5e0955d76 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 @@ -1411,6 +1411,7 @@ class AxolotlInputConfig( "FSDP w/ gradient_accumulation_steps > 1 is broken with transformers==4.46.2. " "Please use a lower value or switch to an older version of transformers." ) + return data class AxolotlConfigWCapabilities(AxolotlInputConfig):