Hi @daemonslayer, please make sure you merge your encoder model & decoder model to a single model and then apply DP on model. If this doesn’t solve your problem, feel free to paste a minimal repro of your script so that we can help.