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-28 20:07:17 +11:00
commit ee0bf6fe66
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
unsloth/save.py
View file

@ -278,7 +278,7 @@ def unsloth_save_model(
not hasattr(internal_model.model, "layers")
):
# Do general saving
print(type(model))
# Edit save_pretrained_settings
# [TODO] _create_repo has errors due to **kwargs getting accepted
for deletion in \
@ -1008,7 +1008,7 @@ def patch_saving_functions(model):
original_model = model
while True:
if not hasattr(original_model, "_original_push_to_hub"):
original_model._original_push_to_hub = original_model.push_to_hub
original_model.push_to_hub = types.MethodType(unsloth_push_to_hub, original_model)