Making the new model work on CPU is sufficient.