From 22abfd6170f4b99a2b1b2822d0b66e187a9fcc51 Mon Sep 17 00:00:00 2001 From: Salman Mohammadi Date: Tue, 18 Mar 2025 11:26:53 +0000 Subject: [PATCH] simplifying check --- src/axolotl/utils/config/models/input/v0_4_1/__init__.py | 1 - 1 file changed, 1 deletion(-) 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 d3a4f873b..674876e52 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 @@ -1689,7 +1689,6 @@ class AxolotlInputConfig( data.get("rl") and data.get("gradient_checkpointing") and data.get("gradient_checkpointing_kwargs") - and "use_reentrant" in data.get("gradient_checkpointing_kwargs") and data.get("gradient_checkpointing_kwargs").get("use_reentrant") and data.get("load_in_4bit") and data.get("adapter") == "qlora"