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-26 03:09:32 -08:00
commit bba36b7d5e
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

@ -1142,14 +1142,14 @@ def unsloth_compile_transformers(
pass
if disable: return
print(model_name)
model_types = get_transformers_model_type(
model_name = model_name,
token = token,
revision = revision,
trust_remote_code = trust_remote_code,
)
print(model_types)
for model_type in model_types:
_unsloth_compile_transformers(
model_type,