search

m1guelpf / yt-whisper

Using OpenAI's Whisper to automatically generate YouTube subtitles
MIT License
1.37k stars 140 forks source link

feat: manual device selection #13

Open ztjhz opened 2 years ago

ztjhz commented 2 years ago

Closes #11

chocies: cuda, cpu

yt_whisper "https://www.youtube.com/watch?v=dQw4w9WgXcQ" --device cpu
yt_whisper "https://www.youtube.com/watch?v=dQw4w9WgXcQ" --device cuda

If --device is not provided, default will be cuda if available else cpu (as implemented in the whisper model).

axfelix commented 2 years ago

Can this be merged? This appears to be defaulting to running on the CPU for me and I'm not sure why...

axfelix commented 2 years ago

Nevermind, I just had to update my pytorch library :)