third_codes/eland
1
0
Fork 0
mirror of https://github.com/elastic/eland.git synced 2025-07-11 00:02:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[DOCS] Adds important note about PyTorch version compatibility. (#487)

Browse Source
This commit is contained in:
István Zoltán Szabó 2022-07-13 12:41:35 +02:00 committed by GitHub
parent c97e69410d
commit fbb01e5698
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -54,6 +54,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.11.0` or earlier to import an NLP model.
Run `pip install torch==1.11` to install the aproppriate version of PyTorch.
### Prerequisites

4
docs/guide/machine-learning.asciidoc
View File

@ -38,6 +38,10 @@ model in {es}.
[[ml-nlp-pytorch]]
=== Natural language processing (NLP) with PyTorch
IMPORTANT: You need to use PyTorch `1.11.0` or earlier to import an NLP model.
Run `pip install torch==1.11` to install the aproppriate version of PyTorch.
For NLP tasks, Eland enables you to import PyTorch trained BERT models into {es}.
Models can be either plain PyTorch models, or supported
https://huggingface.co/transformers[transformers] models from the