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,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