Commit 151b8ed3 authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

repair PLMS

parent b235022c

modules/sd_hijack.py

+3 −1
Original line number Diff line number Diff line
@@ -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

+2 −1
Original line number Diff line number Diff line
@@ -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





def do_inpainting_hijack():

    ldm.models.diffusion.plms.PLMSSampler.p_sample_plms = p_sample_plms

CRA Git | Maintained and supported by SUSTech CRA and CCSE