Skip to content

[BUG]: can not save model in pipeline training mode #6253

Open
@fangxintao

Description

@fangxintao

Is there an existing issue for this bug?

  • I have searched the existing issues

The bug has not been fixed in the latest main branch

  • I have checked the latest main branch

Do you feel comfortable sharing a concise (minimal) script that reproduces the error? :)

Yes, I will share a minimal reproducible script.

🐛 Describe the bug

I use the HybridParallelPlugin methods and set pp_size to 2, tp_size to 1, it can train success, but got the issue
AttributeError: 'Tensor' object has no attribute '_unpad_detach', when I try to use the methods booster.save_model() to save model. I have no ideas to solve this problem. how to deal with this?

Environment

No response

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