Improve PyTorch installation instructions (#677)

This commit is contained in:
Iulia Feroli 2024-03-21 11:21:32 +01:00 committed by GitHub
parent 8e8c49ddbf
commit de83f3f905
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 6 deletions

View File

@ -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

View File

@ -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 <api-key> --url https://<hostname>:<port> ..
--------------------------------------------------
eland_import_hub_model --hub-access-token <access-token> ...
--------------------------------------------------
--
--