From 1d1ce4e5f4c4131a3220baea6ff251fe744b2edb Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Wed, 22 May 2024 10:53:40 +0800
Subject: [PATCH] Update ddpm.py

fix bug
---
 custom_net/ddpm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/custom_net/ddpm.py b/custom_net/ddpm.py
index ddb2f18..fcd6bf6 100644
--- a/custom_net/ddpm.py
+++ b/custom_net/ddpm.py
@@ -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