Skip to content

Commit 67c4ea9

Browse files
fix: disable auto lora kernel if dropout nonzero (axolotl-ai-cloud#2655) [skip ci]
* fix: disable auto lora kernel if dropout nonzero * Add comment from PR feedback --------- Co-authored-by: Wing Lian <[email protected]>
1 parent 526ddb8 commit 67c4ea9

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/axolotl/utils/schemas/config.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1345,6 +1345,10 @@ def check_auto_enable_lora_kernels(cls, data):
13451345
):
13461346
return data
13471347

1348+
# Skip if dropout is not 0, as auto enabling it would just disable it during runtime patch checks
1349+
if data.get("lora_dropout") != 0:
1350+
return data
1351+
13481352
# Check multi-GPU compatibility
13491353
capabilities = data.get("capabilities")
13501354
is_multi_gpu = capabilities and capabilities.get("n_gpu", 0) > 1

0 commit comments

Comments
 (0)