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

cannot import name 'is_torch_npu_available' from 'transformers' #3143

Open
LetsSolveAnalytics opened this issue Dec 22, 2024 · 1 comment
Open

Comments

@LetsSolveAnalytics
Copy link

Fix this by changing in sentence_transformers\util.py

replace 'from transformers import is_torch_npu_available' with 'from transformers.utils.import_utils import is_torch_tpu_available'

@tomaarsen
Copy link
Collaborator

Hello!

is_torch_npu_available and is_torch_tpu_available are not equivalent, I'm afraid. I suspect that your transformers version is a bit too old, as it seems like is_torch_npu_available does not yet exist in it. Could you run pip show transformers to see what version you're on?

For reference, Sentence Transformers requires transformers>=4.41.0.

  • Tom Aarsen

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants