add missing import

used later in line 70
This commit is contained in:
Vladimir Mandic
2023-02-08 07:10:13 -05:00
committed by GitHub
parent 3ee9ca5cb0
commit 3ca41dbded

View File

@@ -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()