Ty_Tech/heretic
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: only export tokenizer when exporting full model

Browse Source 
Fixes #143
This commit is contained in:
Philipp Emanuel Weidmann
2026-02-14 09:18:22 +05:30
parent 6017bcd347
commit e7f8be98b7
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/heretic/main.py
-2
View File
@@ -755,7 +755,6 @@ def run():
merged_model.save_pretrained(save_directory) merged_model.save_pretrained(save_directory)
del merged_model del merged_model
empty_cache() empty_cache()
model.tokenizer.save_pretrained(save_directory) model.tokenizer.save_pretrained(save_directory)
print(f"Model saved to [bold]{save_directory}[/].") print(f"Model saved to [bold]{save_directory}[/].")
@@ -813,7 +812,6 @@ def run():
) )
del merged_model del merged_model
empty_cache() empty_cache()
model.tokenizer.push_to_hub( model.tokenizer.push_to_hub(
repo_id, repo_id,
private=private, private=private,