Hi, I also once assumed they should be adapted to a gpu allocation. Maybe it’s because I  miss something. Anyway, the code is working now. In future, modifying their code myself or asking the author directly might be a good idea.
Thanks for your help!!