sam-pointer-bart-base-v0.3 / requirements.txt
ArneBinder's picture
https://github.com/ArneBinder/pie-document-level/pull/312
3133b5e verified
raw
history blame
1.73 kB
# --------- pytorch-ie --------- #
pytorch-ie>=0.29.6,<0.32.0
pie-datasets>=0.10.5,<0.11.0
pie-modules>=0.14.0,<0.15.0
# --------- models -------- #
adapters>=0.1.2,<0.2.0
pytorch-crf~=0.7.2
# --------- retriever -------- #
langchain>=0.3.0,<0.4.0
langchain-core>=0.3.0,<0.4.0
langchain-community>=0.3.0,<0.4.0
# we use QDrant as vectorstore backend
langchain-qdrant>=0.1.0,<0.2.0
qdrant-client>=1.12.0,<2.0.0
# --------- demo -------- #
gradio~=5.4.0
arxiv~=2.1.3
# --------- hydra --------- #
hydra-core>=1.3.0
hydra-colorlog>=1.2.0
hydra-optuna-sweeper>=1.2.0
# --------- loggers --------- #
wandb
# neptune-client
# mlflow
# comet-ml
# tensorboard
# aim
# --------- linters --------- #
pre-commit # hooks for applying linters on commit
black # code formatting
isort # import sorting
flake8 # code analysis
nbstripout # remove output from jupyter notebooks
# --------- others --------- #
pyrootutils # standardizing the project root setup
python-dotenv # loading env variables from .env file
rich # beautiful text formatting in terminal
pytest # tests
pytest-cov # test coverageataset
sh # for running bash commands in some tests
pudb # debugger
tabulate # show statistics as markdown
plotext # show statistics as plots
prettytable # rendering annotated docs as table (demo)
beautifulsoup4 # rendering annotated docs with displacy + highlighted relations (demo)
# 0.26 seems to be broken when used with adapters, see https://github.com/adapter-hub/adapters/issues/748
huggingface_hub<0.26.0 # interaction with HF hub
networkx~=3.2.1 # to handle segmented entities (e.g if HANDLE_PARTS_OF_SAME=True in demo)