We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fc79606 commit 8446b4aCopy full SHA for 8446b4a
src/axolotl/utils/schemas/config.py
@@ -1319,6 +1319,9 @@ def check_multigpu_lora_kernels(cls, data):
1319
@classmethod
1320
def check_auto_enable_lora_kernels(cls, data):
1321
# 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
1325
if data.get("adapter") in ["lora", "qlora"]:
1326
# Skip if already set, using unsloth optimizations, or using 8-bit
1327
unsloth_fields = ["unsloth_lora_mlp", "unsloth_lora_qkv", "unsloth_lora_o"]
0 commit comments