The techniques here are 100% transferable. It would take some work to migrate it to diffusion + images. But if you tuned the input prompt and rejection detector that is fairly trivial work in a few days.