Update code example in readme to use `device` variable instead of hard coded `cuda`

by hl-tburns - opened Jul 23

←

Jul 23

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.

Xenova

Hugging Face TB Research org Jul 23

Thanks! Would you like to open PRs for the other models too?

Xenova changed pull request status to merged Jul 23

Jul 23

Sure, will do!

Xenova

Hugging Face TB Research org Jul 24

Thanks! Other PRs merged too.

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment