mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 04:52:07 +00:00
hook test for pyinstaller
This commit is contained in:
36
build/pyinstaller/hooks/hook-whisper.py
Normal file
36
build/pyinstaller/hooks/hook-whisper.py
Normal file
@@ -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
|
||||
Reference in New Issue
Block a user