Revert "move torch overlay to device config"

This reverts commit 32fbed96091485fae9f7a30e7789db3d1235897f.