Use with MACE-OFF

Thanks for this very neat piece of work and the clean code! I would love to play with this for protein-ligand systems and was wondering how to swap out the MACE-MP force field with MACE-OFF, and if that's even possible. Are there differences in the architectures that would complicate this? Thanks a lot!