add explicit cast where running without autocast causes issues edd58bf vchiley commited on Jun 15, 2023