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-17 21:31:17 -08:00
commit 8b8951d710
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

@ -1084,7 +1084,7 @@ def unsloth_compile_transformers(
sdap_bool_masks : bool = True,
sdpa_gqa_replace : bool = True,
sdpa_disable_compile : bool = True,
remove_causal_masks : bool = True,
disable_causal_masks : bool = True,
import_from_cache : bool = False,
compile_functions : bool = True,
fuse_lm_head : bool = True,
@ -1102,7 +1102,7 @@ def unsloth_compile_transformers(
sdap_bool_masks = sdap_bool_masks,
sdpa_gqa_replace = sdpa_gqa_replace,
sdpa_disable_compile = sdpa_disable_compile,
remove_causal_masks = remove_causal_masks,
disable_causal_masks = disable_causal_masks,
import_from_cache = import_from_cache,
compile_functions = compile_functions,
fuse_lm_head = fuse_lm_head,