fix missing return

2024-11-14 10:14:13 -05:00
parent 082a41af9d
commit 60763b2e61
1 changed files with 1 additions and 0 deletions
--- 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):