Skip to content

mindnlp0.5.0 from sentence_transformers import SentenceTransformer报错 #2197

@moyu026

Description

@moyu026

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions