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

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

diff --git a/custom_net/modules.py b/custom_net/modules.py
index b960f8b..2c186e0 100644
--- a/custom_net/modules.py
+++ b/custom_net/modules.py
@@ -9,7 +9,7 @@ from transformers import CLIPTokenizer, CLIPTextModel
 import torch.nn.functional as F 
 from torchvision import transforms
 import random
-from customnet_util import default, instantiate_from_config
+from .customnet_util import default, instantiate_from_config
 from .util import make_beta_schedule, extract_into_tensor, noise_like
 import clip