Skip to content

Add support for torch.set_default_device when initializing model parameters#1158

Open
fshp971 wants to merge 1 commit intofacebookresearch:mainfrom fshp971:create-parameter-via-torch-empty