mirror of
https://github.com/tiennm99/DocsGPT.git
synced 2026-10-16 00:14:31 +00:00
feat: optimise embeds
This commit is contained in:
1 parent
e0aff39a1b
commit
8380f9bb47
28 files changed
+1992
-349
No files matched your search
@@ -0,0 +1,23 @@
|
||||
import pytest
|
||||
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def _mpnet_tokenizer():
|
||||
"""Real WordPiece tokenizer; skipped when the hub is unreachable."""
|
||||
pytest.importorskip("tokenizers")
|
||||
from tokenizers import Tokenizer
|
||||
|
||||
try:
|
||||
tokenizer = Tokenizer.from_pretrained("sentence-transformers/all-mpnet-base-v2")
|
||||
except Exception as exc: # offline CI
|
||||
pytest.skip(f"tokenizer unavailable: {exc}")
|
||||
tokenizer.no_padding()
|
||||
tokenizer.no_truncation()
|
||||
return tokenizer
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def hf_counter(_mpnet_tokenizer):
|
||||
from application.parser.tokenization import HuggingFaceCounter
|
||||
|
||||
return HuggingFaceCounter(_mpnet_tokenizer, "sentence-transformers/all-mpnet-base-v2")
|
||||
Reference in new issue
Block a user