goldenretriever-core

View on PyPIReverse Dependencies (0)

0.9.0 goldenretriever_core-0.9.0-py3-none-any.whl

Wheel Details

Project: goldenretriever-core
Version: 0.9.0
Filename: goldenretriever_core-0.9.0-py3-none-any.whl
Download: [link]
Size: 105000
MD5: f89b95e103ba8b7b348b0a58e974bcee
SHA256: 568578363ee88a4312b1943365e5c7930552a4f96842da9ef70dc15afcafbba5
Uploaded: 2024-02-07 18:19:09 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: goldenretriever-core
Version: 0.9.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.3,>=2.1)
Requires-Dist: transformers (<4.38,>=4.33)
Requires-Dist: datasets (<2.17,>=2.15)
Requires-Dist: rich (<14.0.0,>=13.0.0)
Requires-Dist: scikit-learn (<1.5,>=1.3)
Requires-Dist: lightning (<2.2,>=2.1)
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.7.4); 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.7.4); 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: 4412 characters]

WHEEL

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

RECORD

Path Digest Size
goldenretriever/__init__.py sha256=2J0eY-oZzTe3QCOiWRZDfUVvmz4HhLKep9nfRDHISG8 116
goldenretriever/version.py sha256=NPPwYIdBJvNtMByldawzGhKIag6_80raycZ4UnpNVcc 468
goldenretriever/callbacks/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/callbacks/base.py sha256=zvPF5svM53anNwFbD0WndJidr0spIQhDe_kSmSrNHW8 5812
goldenretriever/callbacks/evaluation_callbacks.py sha256=4SttKQlxop4dismL0a2qPqWT3A10fKolmWevxa22wqM 8642
goldenretriever/callbacks/prediction_callbacks.py sha256=SIHKIiQwuq91aVFWBOraO2wanGR_GFcYmqJweB1L_f0 8053
goldenretriever/callbacks/training_callbacks.py sha256=b_0fQxaopDTylqcm7JxuPxmZ9sK4AABon2Jr_SXcJIk 8947
goldenretriever/callbacks/utils_callbacks.py sha256=MXQkRWikoAYf2AZoYhO-PoiKPZDUvkeqW2dGNJtta38 10406
goldenretriever/common/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/common/from_config.py sha256=gzzbWpQvaEkyfJYQ6Q8vPCndOFK51qB7N0QtB071R5w 4082
goldenretriever/common/log.py sha256=94If5TROaIPqSwalaDa-4g84YSSfRv5boeuuWa06ins 4457
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=uGShmWMEHGpCJdj_UqW9UzdgN6Xzoepn6R-PNWEZcxg 4541
goldenretriever/common/utils.py sha256=0klpP7G1TBZv15Hr9aufFfvcVPnhOEs5zR2_eznm0JI 20253
goldenretriever/data/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/data/datasets.py sha256=0zQH1hC37cDne-USmDhsz1_mqT-qSdfddKUOvJDJscc 27031
goldenretriever/data/labels.py sha256=goTeOXqKojZmhWat6S5PI0j_y6gECbssAJLIfhW3vzc 11729
goldenretriever/data/utils.py sha256=g8kfy06iTpOtw1cuEtD0LuUmB_l6TZl2mM_uT6bGesw 6009
goldenretriever/data/base/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/data/base/datasets.py sha256=hhhFRdEKwzvS8l8OC7XacnBuRK_RpM07x9Xnqkx26ok 2490
goldenretriever/indexers/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/indexers/base.py sha256=iUsnOjpzf5ZNQT8suM89gknr10FKyzdyqdBAkxCfe0s 16838
goldenretriever/indexers/document.py sha256=Z1oUVp49SsrTAVPxw_uzhSBiiRO_ltkAaK91f4BNAQs 11421
goldenretriever/indexers/faiss.py sha256=IzjnSkhgQdtNgxF8vV_436DhMfC39ldMi5rMPTuEaTI 16725
goldenretriever/indexers/inmemory.py sha256=BEXbH0z9HS6U3n-FvE02eql-ilGWzcBknXFXRwx-qwI 11971
goldenretriever/indexers/voyager.py sha256=5El-JnDNNn0yFKUWI5pEw1Rw98kN8fd1Ml7EMhP_xHA 10743
goldenretriever/lightning_modules/__init__.py sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs 1
goldenretriever/lightning_modules/pl_data_modules.py sha256=upGSNFyWjICIQhdtM4mGcCTJmH7BT1XMFHZgfdCRf0E 4463
goldenretriever/lightning_modules/pl_modules.py sha256=nLs8RJb-N4ZnO5PPBsn-hrsJPMI4PL6wAfsdNa-Z4EE 4244
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=cpzXm65bMkmlHr5ZDtSczpPVVbqHaAClye4GgEfmQeo 23613
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=SDRLEmPJMqRcNeFN2L_1OPEAuCXoC6EfgsC7GIXBedc 8087
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=5LfB9mj325vwAdeFiMykNBNlHnfiHGIw5yYXcO4z114 6992
goldenretriever/serve/server/backend/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
goldenretriever/serve/server/backend/fastapi.py sha256=sSRb-qNnadX12nV0dZuv1OLmhmRcwbam_QlWuQFwZQ0 4556
goldenretriever/serve/server/backend/ray.py sha256=mWGnzcy8vUf_p3jkX4yB1c0CKsF6ot32SiK5apMP2ro 4929
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=-D-M4e1lr2NohiTUB5NXCslCj4wvspO2nk71DcXZKCg 113
goldenretriever/trainer/train.py sha256=tVn_lscyq5kFKdCQVMEQe6mapoOkrbxCOx2BMYZJFaA 43349
goldenretriever_core-0.9.0.dist-info/METADATA sha256=rCdpFRBYx-l0i1LXu8icFLjDZhfFt23JGA1nxU1kuo8 6932
goldenretriever_core-0.9.0.dist-info/WHEEL sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM 92
goldenretriever_core-0.9.0.dist-info/top_level.txt sha256=hboKCjEFkZonCyIqto7d9IFXsMso3SXeHMRf1pbl0_0 16
goldenretriever_core-0.9.0.dist-info/RECORD

top_level.txt

goldenretriever