Merge pull request #143 from BharatSahAIyak/dev

Adding onnx and whisper to main from dev before release
BharatSahAIyak · Aug 28, 2024 · 2e249e4 · 2e249e4
2 parents 81e35d3 + 7c07fa7
commit 2e249e4
Show file tree

Hide file tree

Showing 3 changed files with 46 additions and 33 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -61,6 +61,7 @@ librosa = "^0.10.2.post1"
 bitsandbytes = "^0.43.3"
 onnx = "^1.16.2"
 onnxruntime = "^1.18.1"
+onnxconverter-common = ">=1.14.0"
 
 [build-system]
 requires = ["poetry-core"]

diff --git a/workflow/training/whisper.py b/workflow/training/whisper.py
@@ -128,3 +128,4 @@ def push_to_hub(self, trainer, save_path, hf_token=None):
         trainer.model.push_to_hub(
             save_path, commit_message="pytorch_model.bin upload/update"
         )
+        self.processor.push_to_hub(save_path, token=hf_token)