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

Update save.py

Browse source
This commit is contained in:
Daniel Han-Chen 2024-01-29 03:43:17 +11:00
commit e10e48893b
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
unsloth/save.py
View file

@ -278,6 +278,7 @@ def unsloth_save_model(
not hasattr(internal_model.model, "layers") not hasattr(internal_model.model, "layers")
): ):
# Do general saving # Do general saving
print(type(model))
# Edit save_pretrained_settings # Edit save_pretrained_settings
# [TODO] _create_repo has errors due to **kwargs getting accepted # [TODO] _create_repo has errors due to **kwargs getting accepted
for deletion in \ for deletion in \