Skip to content

加载mobilebert报错keyError #1886

Closed
@xuhangscut

Description

@xuhangscut

Describe the bug/ 问题描述 (Mandatory / 必填)
A clear and concise description of what the bug is.

  • Hardware Environment(Ascend/GPU/CPU) / 硬件环境:

Please delete the backend not involved / 请删除不涉及的后端:
/device ascend/GPU/CPU/kirin/等其他芯片 ascend

  • Software Environment / 软件环境 (Mandatory / 必填):
    -- MindSpore version (e.g., 1.7.0.Bxxx) : 2.3.1
    -- Python version (e.g., Python 3.7.5) : 3.9
    -- OS platform and distribution (e.g., Linux Ubuntu 16.04):
    -- GCC/Compiler version (if compiled from source):

  • Excute Mode / 执行模式 (Mandatory / 必填)(PyNative/Graph):

Please delete the mode not involved / 请删除不涉及的模式:
/mode pynative

To Reproduce / 重现步骤 (Mandatory / 必填)
使用mindnlp/benchmark/GLUE-QNLI/model_QNLI.py加载google/mobilebert-uncased模型报错
tokenizer = AutoTokenizer.from_pretrained("google/mobilebert-uncased") model=AutoModelForSequenceClassification.from_pretrained("gokuls/mobilebert_sa_GLUE_Experiment_qnli_128", num_labels=2)

Expected behavior / 预期结果 (Mandatory / 必填)
A clear and concise description of what you expected to happen.

Screenshots/ 日志 / 截图 (Mandatory / 必填)
If applicable, add screenshots to help explain your problem.
image
在mindnlp >mindnlp >transformers > models > auto > configuration_auto.py>CONFIG_MAPPING_NAMES 增加了("mobilebert", "MobileBertConfig")配置之后报
image
在如下文件中修改后
image
达到替换 AutoTokenizer直接使用MobileBertTokenizer加载google/mobilebert-uncased相同的报错
Additional context / 备注 (Optional / 选填)
Add any other context about the problem here.

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