Comment by tsadoq

planning to update it to be able to run on it. It's just a matter of finding the keys in the layer dict of the model.