Skip to content

Commit

Permalink
remove additional line (NVIDIA#7293)
Browse files Browse the repository at this point in the history
Signed-off-by: Jason Wang <[email protected]>
Signed-off-by: dorotat <[email protected]>
  • Loading branch information
blahBlahhhJ authored and dorotat-nv committed Aug 24, 2023
1 parent 769b740 commit 0f0bb53
Showing 1 changed file with 0 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -93,7 +93,6 @@ def get_peft_state_dict(self,):
# state_dict keys needs to be in non-O2 format and will be corrected in PEFTSaveRestoreConnector if O2=True
new_k = k.replace("model.module.", "model.", 1)
peft_state_dict[new_k] = state_dict[k]
peft_state_dict[k] = state_dict[k]
return peft_state_dict

def state_dict(self, destination=None, prefix=None, keep_vars=False):
Expand Down

0 comments on commit 0f0bb53

Please sign in to comment.