Update save.py

This commit is contained in:
Daniel Han-Chen 2024-02-20 17:12:18 +11:00
commit 4c9be6d057

View file

@ -174,7 +174,6 @@ def unsloth_save_model(
pass
save_pretrained_settings = dict(locals())
print(save_pretrained_settings)
for deletion in ("model", "tokenizer", "save_method", "temporary_location", "maximum_memory_usage"):
del save_pretrained_settings[deletion]
pass