Officially supported AllenNLP models.
allennlp-models is available on PyPI. To install with
pip, just run
pip install --pre allennlp-models
Note that the
allennlp-models package is tied to the
allennlp core package. Therefore when you install the models package you will get the corresponding version of
allennlp (if you haven't already installed
allennlp). For example,
pip install allennlp-models==1.0.0rc3 pip freeze | grep allennlp # > allennlp==1.0.0rc3 # > allennlp-models==1.0.0rc3
If you intend to install the models package from source, then you probably also want to install
allennlp from source.
Once you have
allennlp installed, run the following within the same Python environment:
git clone https://github.com/allenai/allennlp-models.git cd allennlp-models ALLENNLP_VERSION_OVERRIDE='allennlp' pip install -e . pip install -r dev-requirements.txt
ALLENNLP_VERSION_OVERRIDE environment variable ensures that the
allennlp dependency is unpinned so that your local install of
allennlp will be sufficient. If, however, you haven't installed
allennlp yet and don't want to manage a local install, just omit this environment variable and
allennlp will be installed from the master branch on GitHub.
allennlp-models are developed and tested side-by-side, so they should be kept up-to-date with each other. If you look at the GitHub Actions workflow for
allennlp-models, it's always tested against the master branch of
allennlp is always tested against the master branch of
Docker provides a virtual machine with everything set up to run AllenNLP-- whether you will leverage a GPU or just run on a CPU. Docker provides more isolation and consistency, and also makes it easy to distribute your environment to a compute cluster.
To build an image locally from a specific release, run
docker build \ --build-arg ALLENNLP_VERSION=1.0.0rc3 \ -t allennlp/models - < Dockerfile.release
Just replace "1.0.0rc3" with the desired version.
Alternatively, you can build against specific commits of
docker build \ --build-arg ALLENNLP_COMMIT=e3d72fcb1664caf9554ef4e611191c33a7a5cbbd \ --build-arg ALLENNLP_MODELS_COMMIT=54a5df89da64d8d3869e746bc6dab940552dbfc4 \ -t allennlp/models - < Dockerfile.commit
Just change the
ALLENNLP_MODELS_COMMIT build args to the desired commit SHAs.
Now run the following command to get an environment that will run on either the cpu or gpu.
mkdir -p $HOME/.allennlp/ docker run --rm -v $HOME/.allennlp:/root/.allennlp allennlp/models
If you have GPUs available, you can install the nvidia-docker runtime and then add the flag
--gpus all right before