fixing some issues with our support for 70/405B models#941

HDCharles merged 1 commit intomainpytorch/ao:mainfrom

051_cudamode_irl_kernels_PRpytorch/ao:051_cudamode_irl_kernels_PRCopy head branch name to clipboard