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 22:01:20 -08:00
commit 363ad32293
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
unsloth/models/_utils.py
View file

@ -1095,8 +1095,8 @@ def unsloth_compile_transformers(
revision = revision,
trust_remote_code = trust_remote_code,
)
print(f"Unsloth: Automatic compiler will now patch {model_types}")
for model_type in model_types:
print("Unsloth: Automatic compiler will now patch {model_type}")
_unsloth_compile_transformers(
model_type = model_type,
sdpa_causal_only = sdpa_causal_only,