how to run on GPU on MAC pro

johnma2006 / mamba-minimal

Simple, minimal implementation of the Mamba SSM in one file of PyTorch.

Apache License 2.0

2.62k stars 191 forks source link

Open taozhiyuai opened 2 months ago

taozhiyuai commented 2 months ago

I run demo successfully on my Mac Pro. it seems it runs on CPU only, not GPU. is that possible run on GPU?