goldenretriever-core

View on PyPIReverse Dependencies (1)

1.0.0 goldenretriever_core-1.0.0-py3-none-any.whl

Wheel Details

Project: goldenretriever-core
Version: 1.0.0
Filename: goldenretriever_core-1.0.0-py3-none-any.whl
Download: [link]
Size: 125881
MD5: d7499ee1e5ca822123da05dfe19daaae
SHA256: 34eacfd41ddbef43e9ed13443f24b52e2be756e56497ffa4da9d76f3171cd273
Uploaded: 2024-07-09 12:22:51 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: goldenretriever-core
Version: 1.0.0
Summary: Dense Retriever
Author: Riccardo Orlando
Author-Email: orlandorcc[at]gmail.com
Home-Page: https://github.com/Riccorl/golden-retriever
License: Apache
Keywords: NLP deep learning transformer pytorch retriever rag dpr
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.10
Requires-Dist: torch (<2.4,>=2.3)
Requires-Dist: transformers (<4.42,>=4.41)
Requires-Dist: datasets (<2.20,>=2.19)
Requires-Dist: rich (<14.0.0,>=13.0.0)
Requires-Dist: scikit-learn (<1.5,>=1.3)
Requires-Dist: typer (<0.13,>=0.12)
Requires-Dist: mosaicml (<0.23,>=0.22)
Requires-Dist: lightning (<2.4,>=2.2)
Requires-Dist: mosaicml-streaming (<0.8,>=0.7.5)
Requires-Dist: hydra-core (<1.4,>=1.3)
Requires-Dist: hydra-colorlog
Requires-Dist: wandb (<0.17,>=0.16)
Requires-Dist: art (==6.1)
Requires-Dist: pprintpp (==0.4.0)
Requires-Dist: colorama (==0.4.6)
Requires-Dist: faiss-cpu (==1.8.0); extra == "all"
Requires-Dist: optimum[onnxruntime] (<1.17,>=1.14); extra == "all"
Requires-Dist: termcolor (<2.5,>=2.3); extra == "all"
Requires-Dist: fastapi (<0.110,>=0.104); extra == "all"
Requires-Dist: uvicorn[standard] (<0.28,>=0.24); extra == "all"
Requires-Dist: gunicorn (==21.2.0); extra == "all"
Requires-Dist: ray[serve] (<=2.10,>=2.8); extra == "all"
Requires-Dist: pre-commit; extra == "all"
Requires-Dist: black[d]; extra == "all"
Requires-Dist: isort; extra == "all"
Requires-Dist: termcolor (<2.5,>=2.3); extra == "all-gpu"
Requires-Dist: optimum[onnxruntime-gpu] (<1.8,>=1.7); platform_system == "Linux" and extra == "all-gpu"
Requires-Dist: pre-commit; extra == "dev"
Requires-Dist: black[d]; extra == "dev"
Requires-Dist: isort; extra == "dev"
Requires-Dist: faiss-cpu (==1.8.0); extra == "faiss"
Requires-Dist: optimum[onnxruntime] (<1.17,>=1.14); extra == "onnx"
Requires-Dist: termcolor (<2.5,>=2.3); extra == "onnx"
Requires-Dist: termcolor (<2.5,>=2.3); extra == "onnx-gpu"
Requires-Dist: optimum[onnxruntime-gpu] (<1.8,>=1.7); platform_system == "Linux" and extra == "onnx-gpu"
Requires-Dist: fastapi (<0.110,>=0.104); extra == "serve"
Requires-Dist: uvicorn[standard] (<0.28,>=0.24); extra == "serve"
Requires-Dist: gunicorn (==21.2.0); extra == "serve"
Requires-Dist: ray[serve] (<=2.10,>=2.8); extra == "serve"
Provides-Extra: all
Provides-Extra: all-gpu
Provides-Extra: dev
Provides-Extra: faiss
Provides-Extra: onnx
Provides-Extra: onnx-gpu
Provides-Extra: serve
Description-Content-Type: text/markdown
[Description omitted; length: 4926 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.41.2)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
goldenretriever/__init__.py sha256=2J0eY-oZzTe3QCOiWRZDfUVvmz4HhLKep9nfRDHISG8 116
goldenretriever/__main__.py sha256=WGEoNh7ezhqsCPuWRAyFWqOPonYYcybjg0rW2s4AHDo 75
goldenretriever/version.py sha256=Obhy9q9zlff4EW7anKDAnhVdFprma5xFSbqYrPYQe_w 468
goldenretriever/callbacks/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/callbacks/base.py sha256=ekVxkQN8H0B-W4WGM1SCee_eWY-yhLPc3wZJvh8y_F0 8687
goldenretriever/callbacks/evaluation_callbacks.py sha256=CW7bCjt9J26pcOYPIWRXeJDShHIXGWoGkHLGayojGEc 8997
goldenretriever/callbacks/prediction_callbacks.py sha256=rSshgKhwScq2jvuoOVB3SEPMiKscbHmBNFj0TNYM_zI 11054
goldenretriever/callbacks/training_callbacks.py sha256=rFO2MkU9368oizcrK8BAqRqU7wpSH3u6CejYYJhSX00 12999
goldenretriever/callbacks/utils_callbacks.py sha256=LGwUU4MeXPTfC3TNiOl2vy_Q4UVFW8QB45GVSKmPgSI 27198
goldenretriever/cli/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/cli/cli.py sha256=nG6hUhmcu1gGcJlce5DyXGIQkfBCgHGRkwbt4OQR0TE 776
goldenretriever/cli/utils.py sha256=957EybxIwenqyiXqac2HVwM4TxQMXvzRSDQBJjaHUO0 1412
goldenretriever/common/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/common/data_utils.py sha256=e-fLbxIIAZk4W26OLM1aQBkBdvzQOhK71Xetpq27dkE 8919
goldenretriever/common/dist_utils.py sha256=JlZidorztNkLotaLWuTEeSfZhIlHrU3Gx4cj4Nv8zfE 7308
goldenretriever/common/from_config.py sha256=gzzbWpQvaEkyfJYQ6Q8vPCndOFK51qB7N0QtB071R5w 4082
goldenretriever/common/hf_utils.py sha256=Rud1qmClF7DDTtwyfICCErWTTZhdJsWC1fKiEASyrfY 3977
goldenretriever/common/log.py sha256=MFmnn9m3iZnRl9CBMBcDPuqOtqVBwbl9IU-Mi1QniWI 4559
goldenretriever/common/model_inputs.py sha256=KiulQAPsUGSYGckS8E_XquxziKpjiURxj9BV92QBh6s 1569
goldenretriever/common/sampler.py sha256=F8hFSVAQxry4jm1xk0EdtNKTzKMwbqbhyz0wwQQ3YIU 4030
goldenretriever/common/torch_utils.py sha256=Fn9nU_1p5n4wdsAnBjkeqnlj3nNZPBND4HvVwab2iLE 827
goldenretriever/common/upload.py sha256=Qu8pNEGeQOCv21oZn6ksDyuaR4a-vtAYzclE0EokagM 4515
goldenretriever/common/utils.py sha256=npzFceoL-RcjyDy03Qj-mvr5yY_7bRrBOMTBc9UTDeo 20263
goldenretriever/data/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/data/datasets.py sha256=0j-hUoZKnLQTRThQNa_nMHkQrTZZ_csFXz4wnKc2S0k 32430
goldenretriever/data/labels.py sha256=goTeOXqKojZmhWat6S5PI0j_y6gECbssAJLIfhW3vzc 11729
goldenretriever/data/utils.py sha256=36mzwpmR2fLyuXaSE1vgag-hwnNi34tifHlgT08n9RY 19801
goldenretriever/data/base/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/data/base/datasets.py sha256=f8Asc25Y97lLavz2bEMHqmqT6ucIiA5DjWMu8GOnxIQ 2498
goldenretriever/indexers/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/indexers/base.py sha256=r2rl_yk59owW2wB6T3fBTqD6FtS4GGEU2IkOLSbkL-E 17681
goldenretriever/indexers/document.py sha256=XB6j3n9lt69dJ8w6JWj6bAg2Ovgw467_C0oQhUreH-Y 12275
goldenretriever/indexers/faiss.py sha256=1DPtpPH5RcB8S06M_yr20JBuRJKuDAcnCZpHfCyUzlI 16962
goldenretriever/indexers/inmemory.py sha256=UCEvemiQqDzBcdWEN4x5-PrtQM1V4kWUfnNjU1X2Kck 15477
goldenretriever/lightning_modules/__init__.py sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs 1
goldenretriever/lightning_modules/pl_data_modules.py sha256=TPJIL03g8FD6Zw-svT3ijM_8Ve1FDZcU4WTGDzb5jw8 18251
goldenretriever/lightning_modules/pl_modules.py sha256=PGlRakd0togCJzZx6or77EftF87rOp150WjYMg7kyEw 10958
goldenretriever/pytorch_modules/__init__.py sha256=lWwOUc8LLlu84aQC70gGA1tEkdiIaCw9bWfDR1sCW7o 731
goldenretriever/pytorch_modules/hf.py sha256=Nl1uythP8gULSS9111yUdPekbbxuj-C9u21F3uM3A3U 3727
goldenretriever/pytorch_modules/loss.py sha256=ZgvqL6FCbBPm461X0M25nIxw9-UXycpLv6-qLWXiMtc 1153
goldenretriever/pytorch_modules/model.py sha256=-T7ndrS8Tqm1bQoo948qq-62G7QDDaN0dQw9G8nqiGI 24943
goldenretriever/pytorch_modules/modules.py sha256=NHiFVV4bvH4zIhfA48cYM-R_MbqFX-dUbkT9kc097Y4 329
goldenretriever/pytorch_modules/optim.py sha256=JhHhmSrSd6nG9ub95jMD--suCUzIXHzznyYo9eNqylY 4338
goldenretriever/pytorch_modules/scheduler.py sha256=SkQNT_50oGjMZE8B5F2tFgSGGWZWo8BlNhnYJ9wKcpo 1762
goldenretriever/serve/__init__.py sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs 1
goldenretriever/serve/data/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/serve/data/objects.py sha256=2BUBPaDZ6IJwOLaK0-kB8QRm2g-WmIE2Aq2yLHGb6Ws 1643
goldenretriever/serve/data/splitters/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/serve/data/splitters/base_sentence_splitter.py sha256=JdR1WbxLHdC67zFURbXPISpS8d6oRmIkH4QJk2Y2eGw 1655
goldenretriever/serve/data/splitters/spacy_sentence_splitter.py sha256=272G50JEqwLU_n5LGKgQ0HmAJQWTIwgu0Z9cNrJN7SA 5742
goldenretriever/serve/data/splitters/window_based_splitter.py sha256=LtIYoac4fWeGvBa5GCTevTtnrVTmk03neJOkL4ngLPs 2515
goldenretriever/serve/data/tokenizers/__init__.py sha256=e9ZG67RzeKe--xU7qSWo1RWr7yV3auEZaSBnDUR3MhA 3512
goldenretriever/serve/data/tokenizers/base_tokenizer.py sha256=4up9w1dRXXkF9kOYfW0Fc5-KwWL2gUwBbueHSGiZrD4 2578
goldenretriever/serve/data/tokenizers/regex_tokenizer.py sha256=VZcYSfRDwAJiMJaMKPue3Ziepah8QB9CjQFz5iFKews 2482
goldenretriever/serve/data/tokenizers/spacy_tokenizer.py sha256=ZVfaE3etiGMwE872p_OywswYY-aiMRjBC1ALHbjpDR4 8076
goldenretriever/serve/data/tokenizers/whitespace_tokenizer.py sha256=bNQtebERgPK660OU5w20gk8YB_5sKGrAe_ekcSwY6nY 2246
goldenretriever/serve/data/window/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/serve/data/window/manager.py sha256=xJQlYFI9nfNFdOzby4ZcSI6B1E0yWMHUaRM3qlFh8Nc 3483
goldenretriever/serve/server/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/serve/server/ray_old.py sha256=pJcdnOoZHGox4lbfhLUpN2R8QtUFFoeggauZIkeoXcY 6972
goldenretriever/serve/server/backend/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/serve/server/backend/fastapi.py sha256=iPm-yGoxyrl9HucyuoG1TQFQcCKXgY_GWUtKUYITR8s 4536
goldenretriever/serve/server/backend/ray.py sha256=Jnb_XBBUsYbPY-WE_RhkAxLHn1Bs7fohz7SXQkK3qFI 4957
goldenretriever/serve/server/backend/utils.py sha256=AbfVcLcH53E8I5ooyXsfN2pB22XTZU9a_RFIZy1xHog 1724
goldenretriever/serve/server/frontend/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/serve/server/frontend/relik_front.py sha256=u1fonEurb6x8ol6Pmy8A8Pkc6QZR4x_HhNq1DUMzG0E 8309
goldenretriever/serve/server/frontend/utils.py sha256=tcarB9qYIAnDY1Ulbt4vcD9pkqkNBeSGis6bjYlGfi8 4481
goldenretriever/serve/utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/serve/utils/gunicorn_conf.py sha256=MUl2BL_fP0L16GfohVcE8xEQplMgSSKGAXm6ZBbJS1s 1977
goldenretriever/trainer/__init__.py sha256=C39431w2QezBtKKhFMLNavdCaa5lpfoazelL3ueUIn0 50
goldenretriever/trainer/train.py sha256=zfgO1it5ZeGf2f8qtzYp3BLheJYwmf3YMaCr2y5qRU8 36609
goldenretriever/trainer/utils.py sha256=M7LF1-kUURkv2JuzU1U4S6s_jC1Bds_XOQ9k9XRqoYY 1943
goldenretriever_core-1.0.0.dist-info/METADATA sha256=j5ODbkPfxCP7iOrvPxMFMrpdWa-B4XYDUx8erA8z-qc 7564
goldenretriever_core-1.0.0.dist-info/WHEEL sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A 92
goldenretriever_core-1.0.0.dist-info/entry_points.txt sha256=HzTQLziM21IGtE67ITd4nZ_dW0wqtRsbeqshqtP7K6w 65
goldenretriever_core-1.0.0.dist-info/top_level.txt sha256=hboKCjEFkZonCyIqto7d9IFXsMso3SXeHMRf1pbl0_0 16
goldenretriever_core-1.0.0.dist-info/RECORD

top_level.txt

goldenretriever

entry_points.txt

golden-retriever = goldenretriever.cli.cli:app