requirements.txt

################################
###### Core dependencies #######
################################
torch>=1.10.0
torchvision>=0.8.1
cached-path>=1.1.3
fairscale==0.4.6
jsonnet>=0.10.0 ; sys.platform != 'win32'
nltk>=3.6.5
spacy>=2.1.0
numpy>=1.21.4
tensorboardX>=1.2
requests>=2.28
tqdm>=4.62
h5py>=3.6.0
scikit-learn>=1.0.1
scipy>=1.7.3
pytest>=6.2.5
transformers>=4.1
sentencepiece>=0.1.96
dataclasses;python_version<'3.7'
filelock>=3.3
lmdb>=1.2.1
more-itertools>=8.12.0
termcolor==1.1.0
wandb>=0.10.0
huggingface_hub>=0.0.16
dill>=0.3.4
base58>=2.1.1

# sacremoses should be a dependency of transformers, but it is missing, so we add it manually.
sacremoses

# Spacy depends on typer, and typer had a bug. This is how we make sure we get the fixed version of typer.
typer>=0.4.1

# Protobuf is a dependency of wandb and tensorboard, but they are missing this pin.
protobuf>=3.12.0

# We need this for building the Docker image
traitlets>5.1.1

##################################################
###### Extra dependencies for integrations #######
##################################################
# NOTE: we use a special trailing comment on each line to denote which extras
# each package is needed by. For example, checklist is needed by the 'checklist' extra
# that you install with 'pip install allennlp[checklist]'.
checklist>=0.0.11  # needed by: checklist