Welp, I can do inference on the layer model. Behavior is a bit wonky, but at least it's running.