fix: only export tokenizer when exporting full model

Fixes #143
2026-02-14 09:18:22 +05:30
parent 6017bcd347
commit e7f8be98b7
1 changed files with 6 additions and 8 deletions
@@ -755,7 +755,6 @@ def run():
                                merged_model.save_pretrained(save_directory)
                                del merged_model
                                empty_cache()
-
                                model.tokenizer.save_pretrained(save_directory)

                            print(f"Model saved to [bold]{save_directory}[/].")
@@ -813,7 +812,6 @@ def run():
                                )
                                del merged_model
                                empty_cache()
-
                                model.tokenizer.push_to_hub(
                                    repo_id,
                                    private=private,