add missing import

used later in line 70
YenCheChangGoogle · Feb 8, 2023 · 3ca41db · 3ca41db
1 parent 3ee9ca5
commit 3ca41db
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/modules/sd_hijack_inpainting.py b/modules/sd_hijack_inpainting.py
@@ -11,6 +11,7 @@
 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()