hl-tburns's picture
Update code example in readme to use `device` variable instead of hard coded `cuda`
8d04040 verified