Supports MacOS in MiniCPM-o 2.6

#19
by wanderor - opened

Decouples from cuda.

Note: verified in torch 2.5.1. It does not work in torch 2.3.1 on MacOS (specified in requirements).

bokesyo pinned discussion
OpenBMB org

Thank you for your contribution! Because we only have Mac with 16GB unified memory, so we can't run it on MacOS. However, I ran your modified modeling_minicpmo.py on Linux+CUDA platforms, and there is no problem, and I am happy to merge this PR into main branch, thank you again!

And could you please also add some usage details in README.md, for example, Mac users won't run the code model.cuda() and they need to do model.to('mps')? So that other Mac users can replicate your code, thank you!

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment