search

facebookresearch / co-tracker

CoTracker is a model for tracking any point (pixel) on a video.
https://co-tracker.github.io/
Other
2.52k stars 177 forks source link

How to get weights for CPU support? (M1 Mac support) #42

Closed bharat787 closed 8 months ago

bharat787 commented 8 months ago

Hi,

Thanks for the great work! I was trying to implement the SAM-PT project with co-tracker and downloaded the co-tracker model weights. But I keep running into the error AssertionError: Torch not compiled with CUDA enabled on my M1 MacBook machine.

Is there a way to get the weights running on for CPU-only support? They seem to be hardcoded. (I did go through issue #9, but the suggestion there seems to be if I'm running co-tracker demo on local, although here for SAM-PT, I'm just trying to use the weights).

Thanks for the help!

nikitakaraevv commented 8 months ago

Hi @bharat787, thank you! I see you've opened a pull request https://github.com/facebookresearch/co-tracker/pull/43. I assume this fixes the problem?

bharat787 commented 8 months ago

Hi, yes it does. Thanks for the merge!