Move to cpu when computing firing counts#174

SrGonao merged 1 commit intomainEleutherAI/delphi:mainfrom

pr/latents-cache-cleanupEleutherAI/delphi:pr/latents-cache-cleanupCopy head branch name to clipboard