first commit

lib/attention.py

@@ -18,7 +18,7 @@ from torch import nn, einsum
 from einops import rearrange, repeat
 from typing import Optional, Any
 
-from utils import checkpoint
+from util import checkpoint
 
 try:
     import xformers

utils.py

@@ -19,6 +19,7 @@ from inspect import isfunction
 from PIL import Image, ImageDraw, ImageFont
 
 
+
 def log_txt_as_img(wh, xc, size=10):
     # wh a tuple of (width, height)
     # xc a list of captions to plot
@@ -114,4 +115,4 @@ def create_model(config_path):
     config = OmegaConf.load(config_path)
     model = instantiate_from_config(config.model).cpu()
     print(f'Loaded model config from [{config_path}]')
-    return model
+    return model