My solution to that was simply to add this model as an optional output transformer: https://steganography.live/info