-
Notifications
You must be signed in to change notification settings - Fork 256
Description
Describe the bug/ 问题描述 (Mandatory / 必填)
import mindnlp
from sentence_transformers import SentenceTransformer报错
-
Hardware Environment(
Ascend/GPU/CPU) / 硬件环境:Ascend -
Software Environment / 软件环境 (Mandatory / 必填):
-- MindSpore version (e.g., 1.7.0.Bxxx) :2.7
-- Python version (e.g., Python 3.7.5) :3.10
-- OS platform and distribution (e.g., Linux Ubuntu 16.04):Ubuntu 22 -
Excute Mode / 执行模式 (Mandatory / 必填)(
PyNative/Graph):Graph
Please delete the mode not involved / 请删除不涉及的模式:
/mode pynative
/mode graph
To Reproduce / 重现步骤 (Mandatory / 必填)
git clone https://gitee.com/mindspore-lab/mindnlp.git
cd mindnlp
git checkout 6719e5e
bash scripts/build_and_reinstall.sh
import mindnlp
from sentence_transformers import SentenceTransformer
Screenshots/ 日志 / 截图 (Mandatory / 必填)
Traceback (most recent call last):
File "/home/ma-user/work/rag_test/mindnlp/llm/rag/startup.py", line 26, in
from embedding import EmbeddingsFunAdapter
File "/home/ma-user/work/rag_test/mindnlp/llm/rag/embedding.py", line 23, in
from sentence_transformers import SentenceTransformer
File "/usr/local/python3.10.14/lib/python3.10/site-packages/sentence_transformers/init.py", line 15, in
from sentence_transformers.cross_encoder import (
File "/usr/local/python3.10.14/lib/python3.10/site-packages/sentence_transformers/cross_encoder/init.py", line 5, in
from .trainer import CrossEncoderTrainer
File "/usr/local/python3.10.14/lib/python3.10/site-packages/sentence_transformers/cross_encoder/trainer.py", line 18, in
from sentence_transformers.cross_encoder.losses import BinaryCrossEntropyLoss, CrossEntropyLoss
File "/usr/local/python3.10.14/lib/python3.10/site-packages/sentence_transformers/cross_encoder/losses/init.py", line 4, in
from .CachedMultipleNegativesRankingLoss import CachedMultipleNegativesRankingLoss
File "/usr/local/python3.10.14/lib/python3.10/site-packages/sentence_transformers/cross_encoder/losses/CachedMultipleNegativesRankingLoss.py", line 10, in
from torch.utils.checkpoint import get_device_states, set_device_states
File "/usr/local/python3.10.14/lib/python3.10/site-packages/mindnlp/utils/torch_proxy.py", line 29, in find_spec
raise e
File "/usr/local/python3.10.14/lib/python3.10/site-packages/mindnlp/utils/torch_proxy.py", line 27, in find_spec
importlib.import_module(target_name)
File "/usr/local/python3.10.14/lib/python3.10/importlib/init.py", line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
ModuleNotFoundError: No module named 'mindnlp.core.utils.checkpoint.get_device_states'; 'mindnlp.core.utils.checkpoint' is not a package