Is it possible to run this on new models? It seem like the code is only for inference, unless I’m misunderstanding