yt-whisper icon indicating copy to clipboard operation
yt-whisper copied to clipboard

Published 20 hours ago • verified publisher icon m1guelpf

feat: manual device selection

Open ztjhz opened this issue 2 years ago • 2 comments

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).

ztjhz avatar Sep 30 '22 05:09 ztjhz

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

axfelix avatar Oct 19 '22 20:10 axfelix

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

axfelix avatar Oct 19 '22 21:10 axfelix