Update ddpm.py

fix bug
This commit is contained in:
smthemex
2024-05-22 10:53:40 +08:00
committed by GitHub
parent 3375171fb3
commit 1d1ce4e5f4
+1 -1
View File
@@ -25,7 +25,7 @@ from .ema import LitEma
from .distributions import normal_kl, DiagonalGaussianDistribution
from .autoencoder import VQModelInterface, IdentityFirstStage, AutoencoderKL
from .util import make_beta_schedule, extract_into_tensor, noise_like
from .ddim import DDIMSampler
from .dddim import DDIMSampler
from .attention import CrossAttention