diff --git a/build/pyinstaller/hooks/hook-whisper.py b/build/pyinstaller/hooks/hook-whisper.py new file mode 100644 index 00000000000..f0b3e474361 --- /dev/null +++ b/build/pyinstaller/hooks/hook-whisper.py @@ -0,0 +1,36 @@ +import importlib + +import importlib_metadata +from PyInstaller.utils.hooks import copy_metadata + +datas = copy_metadata("transformers") +datas += copy_metadata("tokenizers") +datas += copy_metadata("tqdm") +datas += copy_metadata("regex") +datas += copy_metadata("requests") +datas += copy_metadata("packaging") +datas += copy_metadata("filelock") +datas += copy_metadata("numpy") +datas += copy_metadata("dataclasses") +datas += copy_metadata("torch") + +candidates = [ + "tensorflow", + "tensorflow-cpu", + "tensorflow-gpu", + "tf-nightly", + "tf-nightly-cpu", + "tf-nightly-gpu", + "intel-tensorflow", + "intel-tensorflow-avx512", + "tensorflow-rocm", + "tensorflow-macos", + "tensorflow-aarch64", +] +for candidate in datas: + try: + if importlib.util.find_spec(candidate): + datas += copy_metadata(candidate) + break + except importlib_metadata.PackageNotFoundError: + pass