From ad0c2bcd878bc6b0395ad9d83d3ef25d5ac5927c Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Mon, 19 Feb 2024 13:08:03 +1100 Subject: [PATCH] Update save.py --- unsloth/save.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/unsloth/save.py b/unsloth/save.py index 81e7a487fc..84a0745e4d 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -441,9 +441,10 @@ def unsloth_save_model( commit_description += " (Trained with Unsloth 2x faster)" pass + # commit_description does not seem to work? what_to_delete = ("use_temp_dir", "commit_message", "create_pr", "revision", "commit_description", "tags",) \ if username is None else \ - ("use_temp_dir", "create_pr", "revision", "tags",) + ("use_temp_dir", "create_pr", "revision", "tags", "commit_description",) for deletion in what_to_delete: del save_pretrained_settings[deletion] pass