Update modules.py

2024-08-11 16:22:26 +08:00
parent 584be35c65
commit b78d68ccfe
1 changed files with 1 additions and 1 deletions
@@ -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