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-02-18 23:35:18 +11:00
commit 89d2418cd7
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/save.py
View file

@ -1045,9 +1045,9 @@ def patch_saving_functions(model):
original_model = model original_model = model
while True: while True:
if hasattr(original_model, "original_push_to_hub"): if hasattr(original_model, "original_push_to_hub"):
original_model.push_to_hub = original_model.original_push_to_hub, original_model original_model.push_to_hub = original_model.original_push_to_hub
pass pass
original_model.original_push_to_hub = original_model.push_to_hub, original_model original_model.original_push_to_hub = original_model.push_to_hub
if hasattr(original_model, "model"): original_model = original_model.model if hasattr(original_model, "model"): original_model = original_model.model
else: break else: break
@ -1105,7 +1105,7 @@ def patch_saving_functions(model):
while True: while True:
if not hasattr(original_model, "original_push_to_hub"): if not hasattr(original_model, "original_push_to_hub"):
original_model.original_push_to_hub = original_model.push_to_hub, original_model original_model.original_push_to_hub = original_model.push_to_hub
pass pass
original_model.push_to_hub = types.MethodType(unsloth_push_to_hub, original_model) original_model.push_to_hub = types.MethodType(unsloth_push_to_hub, original_model)