Skip to content

Commit 8446b4a

Browse files
authored
don't automatically enable lora kernels for RL training (axolotl-ai-cloud#2600)
1 parent fc79606 commit 8446b4a

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/axolotl/utils/schemas/config.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1319,6 +1319,9 @@ def check_multigpu_lora_kernels(cls, data):
13191319
@classmethod
13201320
def check_auto_enable_lora_kernels(cls, data):
13211321
# Only proceed if using LoRA or QLoRA adapter
1322+
if data.get("rl"):
1323+
# RL trainers not tested so don't enable kernels by default
1324+
return data
13221325
if data.get("adapter") in ["lora", "qlora"]:
13231326
# Skip if already set, using unsloth optimizations, or using 8-bit
13241327
unsloth_fields = ["unsloth_lora_mlp", "unsloth_lora_qkv", "unsloth_lora_o"]

0 commit comments

Comments
 (0)