add missing import
used later in line 70
This commit is contained in:
parent
3ee9ca5cb0
commit
3ca41dbded
@ -11,6 +11,7 @@ import ldm.models.diffusion.plms
|
|||||||
from ldm.models.diffusion.ddpm import LatentDiffusion
|
from ldm.models.diffusion.ddpm import LatentDiffusion
|
||||||
from ldm.models.diffusion.plms import PLMSSampler
|
from ldm.models.diffusion.plms import PLMSSampler
|
||||||
from ldm.models.diffusion.ddim import DDIMSampler, noise_like
|
from ldm.models.diffusion.ddim import DDIMSampler, noise_like
|
||||||
|
from ldm.models.diffusion.sampling_util import norm_thresholding
|
||||||
|
|
||||||
|
|
||||||
@torch.no_grad()
|
@torch.no_grad()
|
||||||
|
Loading…
Reference in New Issue
Block a user