chenxwh / cog-whisper

MIT License
81 stars 28 forks source link

Upgrade cuda to upgrade ubuntu to upgrade ffmpeg #5

Closed anotherjesse closed 1 year ago

anotherjesse commented 2 years ago

Using this patch relies on this PR being included in your cog binary

https://github.com/replicate/cog/pull/779

The version of CUDA changes the docker image that is used as a base.

nickstenning commented 2 years ago

Note that this fixes #2

maccman commented 1 year ago

Would love to get this merged :)

chenxwh commented 1 year ago

Thanks for your suggestion

@bfirsh I tried it does not seem Cuda 11.7 works with Cog yet? Maybe I am wrong but with the same cog.yaml otherwise cuda 11.6 works

bfirsh commented 1 year ago

@chenxwh you probably need to upgrade Cog.

bfirsh commented 1 year ago

In theory with latest version of Cog, we can just specify torch==1.10 in python_packages and leave out cuda entirely and it'll all just work.

This PR also probably won't work direct -- there is a hack that hard codes Torch compiled against CUDA 11.3. But if we just remove the hack it should all just resolve automatically.

chenxwh commented 1 year ago

I have pushed a version with torch=1.13.0 and NOT specifying cuda version, could you try if it works for your .mp4 input @maccman @anotherjesse?

maccman commented 1 year ago

Okay will try!

anotherjesse commented 1 year ago

My sample worked! The replicate version now works with iPhone / macOS voice recorder