diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index e1fa66fe32..5cdb67cfd3 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -1089,7 +1089,7 @@ def unsloth_compile_transformers( sdpa_dynamic_mask = True, sdpa_bool_masks = True, sdpa_gqa_replace = True, - sdpa_disable_compile = True, + sdpa_dynamic_compile = True, compile_attention = True, disable_causal_masks = True, compile_torch_modules = True, @@ -1116,7 +1116,7 @@ def unsloth_compile_transformers( sdpa_dynamic_mask = sdpa_dynamic_mask, sdpa_bool_masks = sdpa_bool_masks, sdpa_gqa_replace = sdpa_gqa_replace, - sdpa_disable_compile = sdpa_disable_compile, + sdpa_dynamic_compile = sdpa_dynamic_compile, compile_attention = compile_attention, disable_causal_masks = disable_causal_masks, compile_torch_modules = compile_torch_modules,