Merge pull request #7651 from vladmandic/missing-imports
Add missing imports
This commit is contained in:
commit
b63a13c5ed
|
@ -1,5 +1,6 @@
|
||||||
# this file is adapted from https://github.com/victorca25/iNNfer
|
# this file is adapted from https://github.com/victorca25/iNNfer
|
||||||
|
|
||||||
|
from collections import OrderedDict
|
||||||
import math
|
import math
|
||||||
import functools
|
import functools
|
||||||
import torch
|
import torch
|
||||||
|
|
|
@ -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