Does the method work on other diffusion models?

[Nattograpes]

Great work!
I wonder if the method works on the sd2.1 or other version, which is helpful for the extended application.