repair PLMS

modules/sd_hijack.py

@@ -6,7 +6,7 @@ import modules.textual_inversion.textual_inversion
 from modules import devices, sd_hijack_optimizations, shared, script_callbacks, errors, sd_unet
 from modules.hypernetworks import hypernetwork
 from modules.shared import cmd_opts
-from modules import sd_hijack_clip, sd_hijack_open_clip, sd_hijack_unet, sd_hijack_xlmr, xlmr
+from modules import sd_hijack_clip, sd_hijack_open_clip, sd_hijack_unet, sd_hijack_xlmr, xlmr, sd_hijack_inpainting
 
 import ldm.modules.attention
 import ldm.modules.diffusionmodules.model
@@ -35,6 +35,8 @@ ldm.modules.diffusionmodules.model.print = shared.ldm_print
 ldm.util.print = shared.ldm_print
 ldm.models.diffusion.ddpm.print = shared.ldm_print
 
+sd_hijack_inpainting.do_inpainting_hijack()
+
 optimizers = []
 current_optimizer: sd_hijack_optimizations.SdOptimization = None
 

modules/sd_hijack_inpainting.py

@@ -91,4 +91,5 @@ def p_sample_plms(self, x, c, t, index, repeat_noise=False, use_original_steps=F
     return x_prev, pred_x0, e_t
 
 
-ldm.models.diffusion.plms.PLMSSampler.p_sample_plms = p_sample_plms
+def do_inpainting_hijack():
+    ldm.models.diffusion.plms.PLMSSampler.p_sample_plms = p_sample_plms