Skip to content

fix(pipe): make pipe module load_state_dir non-strict-mode work - #4020

Merged
mrwyattii merged 5 commits into
deepspeedai:masterfrom
hughpu:fix/pipe_load_stat
Jul 25, 2023
Merged

fix(pipe): make pipe module load_state_dir non-strict-mode work#4020
mrwyattii merged 5 commits into
deepspeedai:masterfrom
hughpu:fix/pipe_load_stat

Conversation

@hughpu

@hughpu hughpu commented Jul 24, 2023

Copy link
Copy Markdown
Contributor
  • strict parameter of method PipelineModule.load_state_dir was not passed into layer.load_state_dict
  • issue will cause modifed model such as LORA, failed to load checkpoints from pretrained model

@mrwyattii
mrwyattii enabled auto-merge July 25, 2023 17:04
@mrwyattii
mrwyattii added this pull request to the merge queue Jul 25, 2023
Merged via the queue into deepspeedai:master with commit 09601bb Jul 25, 2023
@hughpu
hughpu deleted the fix/pipe_load_stat branch July 30, 2023 04:54
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@ShadenSmith ShadenSmith Awaiting requested review from ShadenSmith
@duli2012 duli2012 Awaiting requested review from duli2012
1 more reviewer
@mrwyattii mrwyattii mrwyattii approved these changes
Reviewers whose approvals may not affect merge requirements

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@hughpu @mrwyattii