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

Question

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

bharat787 opened this issue 10 months ago · comments

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!

Nikita Karaev · Answer 1 · Mon Oct 30 2023 20:05:13 GMT+0800 (China Standard Time)

Hi @bharat787, thank you! I see you've opened a pull request #43.
I assume this fixes the problem?

Bharat Gupta · Answer 2 · Mon Oct 30 2023 20:44:41 GMT+0800 (China Standard Time)

Hi, yes it does. Thanks for the merge!