Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Upgrade PyTorch to version 2.3.1 #718

Merged
merged 6 commits into from
Sep 30, 2024

Conversation

davidkyle
Copy link
Member

@davidkyle davidkyle commented Aug 15, 2024

Elasticsearch has upgraded to PyTorch to 2.3.1 in 8.16 and 8.15.2. This PR updates Eland to use the same version.

The usual compatibility issues apply, Eland will refuse to upload to an Elasticsearch cluster that has is using an earlier version of PyTorch. The range of compatible versions of Sentence Transformers and HuggingFace Transformers has been updated to incorporate the latest release of the same major version.

Elasticsearch is upgraded in elastic/ml-cpp#2705

@davidkyle davidkyle added enhancement New feature or request topic:NLP Issue or PR about NLP model support and eland_import_hub_model labels Aug 15, 2024
Copy link
Member

@pquentin pquentin left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks! LGTM, but we need to make sure this really makes it into 8.15.1 before merging.

eland/cli/eland_import_hub_model.py Outdated Show resolved Hide resolved
eland/cli/eland_import_hub_model.py Outdated Show resolved Hide resolved
Copy link
Member

@pquentin pquentin left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks! LGTM.

@davidkyle davidkyle merged commit 5253501 into elastic:main Sep 30, 2024
4 checks passed
@davidkyle davidkyle deleted the upgrade-pytorch-2.3.1 branch September 30, 2024 09:22
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request topic:NLP Issue or PR about NLP model support and eland_import_hub_model
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants