Running the models on MAC

[sid0710]

Hi, the results looks great, and thanks for sharing the model weights as well. I wanted to know whether it is possible to use the models without the GPUs similar to CLAP? I am trying to run this on M1 Mac.

[gzhu06]

Thank you for your interest in our project, since our models are trained on jax/flax, it's still possible but you may need a lot of hack to make it work, including make jax/flax packages work on CPU and write some code to load these weights into CPU.

You can refer here: https://github.com/google/flax?tab=readme-ov-file#quick-install on how to install CPU version, and ask Claude or ChatGPT to give some code on loading weights to CPU to start with.