fix fsdp loading of models, esp 70b (#1780)

fix fsdp loading of models, esp 70b (#1780) #144