Update code example in readme to use `device` variable instead of hard coded `cuda`
#5
by
hl-tburns
- opened
The code example uses device for model creation but then uses a hard coded cuda for inputs. Updated to use device for inputs as well.
Thanks! Would you like to open PRs for the other models too?
Xenova
changed pull request status to
merged
Sure, will do!
Thanks! Other PRs merged too.