Skip to content

Commit 579557e

Browse files
committed
Bundle transformers model modules for release
1 parent 9f0e4ca commit 579557e

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

scripts/package_release.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,7 @@
1717
ENTRYPOINT = ROOT / "src" / "bitloops_embeddings" / "__main__.py"
1818
PYINSTALLER_COLLECT_SUBMODULES = [
1919
"sentence_transformers",
20-
"transformers.models.auto",
21-
"transformers.models.xlm_roberta",
20+
"transformers.models",
2221
]
2322
PYINSTALLER_COLLECT_DATA = [
2423
"sentence_transformers",

0 commit comments

Comments
 (0)