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
+6-6