add missing import

used later in line 70
3 years ago · 3ca41dbded
parent 3ee9ca5cb0
commit 3ca41dbded
1 changed files with 1 additions and 0 deletions
--- a/modules/sd_hijack_inpainting.py
+++ b/modules/sd_hijack_inpainting.py
@ -11,6 +11,7 @@ import ldm.models.diffusion.plms
 from ldm.models.diffusion.ddpm import LatentDiffusion
 from ldm.models.diffusion.plms import PLMSSampler
 from ldm.models.diffusion.ddim import DDIMSampler, noise_like
+from ldm.models.diffusion.sampling_util import norm_thresholding


@torch.no_grad()