From b50a276b7afa85e9294e50d3dffcb2a4943c6c3a Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Sun, 11 Aug 2024 16:23:01 +0800
Subject: [PATCH] Update ddpm.py

---
 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 fcd6bf6..6f3b9a2 100644
--- a/custom_net/ddpm.py
+++ b/custom_net/ddpm.py
@@ -20,7 +20,7 @@ from torchvision.utils import make_grid
 from pytorch_lightning.utilities.rank_zero import rank_zero_only
 from omegaconf import ListConfig
 
-from customnet_util import log_txt_as_img, exists, default, ismap, isimage, mean_flat, count_params, instantiate_from_config
+from .customnet_util import log_txt_as_img, exists, default, ismap, isimage, mean_flat, count_params, instantiate_from_config
 from .ema import LitEma
 from .distributions import normal_kl, DiagonalGaussianDistribution
 from .autoencoder import VQModelInterface, IdentityFirstStage, AutoencoderKL