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 20:22:52 +11:00
commit 53f6a07b92
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/save.py
View file

@ -1046,7 +1046,11 @@ def patch_saving_functions(model):
while True:
if hasattr(original_model, "original_push_to_hub"):
original_model.push_to_hub = types.MethodType(original_model.original_push_to_hub, original_model)
del original_model.original_push_to_hub
print(original_model)
try:
del original_model.original_push_to_hub
except:
print("!")
if hasattr(original_model, "push_to_hub_merged"): del original_model.push_to_hub_merged
if hasattr(original_model, "save_pretrained_merged"): del original_model.save_pretrained_merged
if hasattr(original_model, "push_to_hub_gguf"): del original_model.push_to_hub_gguf