Update code example in readme to use `device` variable instead of hard coded `cuda`
a62f674
verified
hl-tburns
commited on