fix: adopt latest state_dict processing#286

ruixin31 wants to merge 1 commit intomlfoundations:mainmlfoundations/open_lm:mainfrom

ruixin31:fix-from-pretrainedruixin31/open_lm:fix-from-pretrainedCopy head branch name to clipboard