Skip to content

bloom_560m导出mindir模型报错ValueError #15

@hql-hql

Description

@hql-hql

mindformers版本:1.0.2
模型:bloom_560m
参考:https://github.com/mindspore-lab/mindformers/blob/r1.0/docs/feature_cards/Inference.md
方式3:针对Configs下已支持模型的导出方式(以bloom_560m为例)
run_bloom_560m.yaml已经文档要求修改。
报错:ValueError: For 'Mul', x.shape and y.shape need to broadcast. The value of x.shape[-1] or y.shape[-1] must be 1 or -1 when they are not the same, but got x.shape = [1, 1024] and y.shape = [1, 512].

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions