Hi, If I try to run this with CPU, it raises an error: ``` NotImplementedError('no support for using CPU-only because lazy') ```