hi @juierror, you might take a look at [DISCUSS] Module based Model Runtime Interface
I’m not sure that RFC has been implemented yet, but I think that is the general idea. you might be able to do this by defining a Relay model with two model functions, but I haven’t tried this yet so i’m not sure what challenges you may face if you do this.