UKPLab/sentence-transformers

cannot import name 'is_torch_npu_available' from 'transformers'

Opened this issue · 1 comments

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'

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