ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update _utils.py

Browse source
This commit is contained in:
Daniel Han 2024-11-19 02:25:22 -08:00
commit b3e5f72aea
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
unsloth/models/_utils.py
View file

@ -1081,7 +1081,7 @@ def unsloth_compile_transformers(
revision = None,
trust_remote_code : bool = False,
sdpa_causal_only : bool = False,
sdap_bool_masks : bool = True,
sdpa_bool_masks : bool = True,
sdpa_gqa_replace : bool = True,
sdpa_disable_compile : bool = True,
disable_causal_masks : bool = True,
@ -1100,7 +1100,7 @@ def unsloth_compile_transformers(
_unsloth_compile_transformers(
model_type = model_type,
sdpa_causal_only = sdpa_causal_only,
sdap_bool_masks = sdap_bool_masks,
sdpa_bool_masks = sdpa_bool_masks,
sdpa_gqa_replace = sdpa_gqa_replace,
sdpa_disable_compile = sdpa_disable_compile,
disable_causal_masks = disable_causal_masks,