Optimize GPU checkpoint loading by ensuring model transfer before load_state_dict on build method#325

nimanikoo wants to merge 1 commit intomicrosoft:mainmicrosoft/BitNet:mainfrom

nimanikoo:feature/direct-gpu-checkpoint-loadingnimanikoo/BitNet:feature/direct-gpu-checkpoint-loadingCopy head branch name to clipboard