From e7f8be98b7fbd82393edebcf0d3c77edd964e666 Mon Sep 17 00:00:00 2001 From: Philipp Emanuel Weidmann Date: Sat, 14 Feb 2026 09:18:22 +0530 Subject: [PATCH] fix: only export tokenizer when exporting full model Fixes #143 --- src/heretic/main.py | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/src/heretic/main.py b/src/heretic/main.py index 031ea5e..d626e6c 100644 --- a/src/heretic/main.py +++ b/src/heretic/main.py @@ -755,8 +755,7 @@ def run(): merged_model.save_pretrained(save_directory) del merged_model empty_cache() - - model.tokenizer.save_pretrained(save_directory) + model.tokenizer.save_pretrained(save_directory) print(f"Model saved to [bold]{save_directory}[/].") @@ -813,12 +812,11 @@ def run(): ) del merged_model empty_cache() - - model.tokenizer.push_to_hub( - repo_id, - private=private, - token=token, - ) + model.tokenizer.push_to_hub( + repo_id, + private=private, + token=token, + ) # If the model path doesn't exist locally, it can be assumed # to be a model hosted on the Hugging Face Hub, in which case