ysharma HF staff commited on
Commit
d9d82c7
1 Parent(s): 56826a7
Files changed (1) hide show
  1. ddpm.py +6 -6
ddpm.py CHANGED
@@ -18,12 +18,12 @@ from tqdm import tqdm
18
  from torchvision.utils import make_grid
19
  from pytorch_lightning.utilities.distributed import rank_zero_only
20
 
21
- from ldm.util import log_txt_as_img, exists, default, ismap, isimage, mean_flat, count_params, instantiate_from_config
22
- from ldm.modules.ema import LitEma
23
- from ldm.modules.distributions.distributions import normal_kl, DiagonalGaussianDistribution
24
- from ldm.models.autoencoder import VQModelInterface, IdentityFirstStage, AutoencoderKL
25
- from ldm.modules.diffusionmodules.util import make_beta_schedule, extract_into_tensor, noise_like
26
- from ldm.models.diffusion.ddim import DDIMSampler
27
 
28
 
29
  __conditioning_keys__ = {'concat': 'c_concat',
 
18
  from torchvision.utils import make_grid
19
  from pytorch_lightning.utilities.distributed import rank_zero_only
20
 
21
+ from util import log_txt_as_img, exists, default, ismap, isimage, mean_flat, count_params, instantiate_from_config
22
+ from ema import LitEma
23
+ from distributions import normal_kl, DiagonalGaussianDistribution
24
+ from autoencoder import VQModelInterface, IdentityFirstStage, AutoencoderKL
25
+ from util2 import make_beta_schedule, extract_into_tensor, noise_like
26
+ from ddim import DDIMSampler
27
 
28
 
29
  __conditioning_keys__ = {'concat': 'c_concat',