diff --git a/README.md b/README.md index 180ac3c..ea0ad85 100644 --- a/README.md +++ b/README.md @@ -58,8 +58,8 @@ $ conda install -c conda-forge eland If you are using the NLP with PyTorch feature make sure your Eland minor version matches the minor version of your Elasticsearch cluster. For all other features it is sufficient for the major versions to match. -- You need to use PyTorch `1.13.1` or earlier to import an NLP model. - Run `pip install torch==1.13.1` to install the aproppriate version of PyTorch. +- You need to install the appropriate version of PyTorch to import an NLP model. Run `python -m pip + install 'eland[pytorch]'` to install that version. ### Prerequisites diff --git a/docs/guide/machine-learning.asciidoc b/docs/guide/machine-learning.asciidoc index 5c944d1..eee3b86 100644 --- a/docs/guide/machine-learning.asciidoc +++ b/docs/guide/machine-learning.asciidoc @@ -38,9 +38,8 @@ model in {es}. [[ml-nlp-pytorch]] === Natural language processing (NLP) with PyTorch - -IMPORTANT: You need to use PyTorch `1.13` or earlier to import an NLP model. -Run `pip install torch==1.13` to install the aproppriate version of PyTorch. +IMPORTANT: You need to install the appropriate version of PyTorch to import an +NLP model. Run `python -m pip install 'eland[pytorch]'` to install that version. For NLP tasks, Eland enables you to import PyTorch models into {es}. Use the `eland_import_hub_model` script to download and install supported @@ -191,4 +190,4 @@ eland_import_hub_model --es-api-key --url https://: .. -------------------------------------------------- eland_import_hub_model --hub-access-token ... -------------------------------------------------- --- \ No newline at end of file +--