first commit

lib/attention.py

@@ -8,6 +8,10 @@
  * Copyright (c) 2023 Lvmin Zhang and Maneesh Agrawala
 '''
 
+import sys
+
+sys.path.append('./')
+
 from inspect import isfunction
 import math
 import torch
@@ -16,7 +20,7 @@ from torch import nn, einsum
 from einops import rearrange, repeat
 from typing import Optional, Any
 
-from ..utils import checkpoint
+from utils import checkpoint
 
 try:
     import xformers

lib/util.py

@@ -18,6 +18,10 @@
 # thanks!
 
 
+import sys
+
+sys.path.append('./')
+
 import os
 import math
 import torch
@@ -25,7 +29,7 @@ import torch.nn as nn
 import numpy as np
 from einops import repeat
 
-from ..utils import instantiate_from_config
+from utils import instantiate_from_config
 
 
 def make_beta_schedule(schedule, n_timestep, linear_start=1e-4, linear_end=2e-2, cosine_s=8e-3):