llm-foundry

596dd9dc - Do dtype conversion in torch hook to save memory (#1384)

Do dtype conversion in torch hook to save memory (#1384) * Do dtype conversion in torch hook to save memory * update code comment Co-authored-by: Saaketh Narayan <saaketh@mosaicml.com> --------- Co-authored-by: Saaketh Narayan <saaketh@mosaicml.com>