Update unet/openaimodel.py

unet/openaimodel.py

@@ -8,7 +8,7 @@ import torch as th
 import torch.nn as nn
 import torch.nn.functional as F
 
-from util import (
+from .util import (
     checkpoint,
     conv_nd,
     linear,
@@ -16,13 +16,14 @@ from util import (
     zero_module,
     normalization,
     timestep_embedding,
+    exists
 )
 
 # replace with custom transformer
-from mv_attention import SPADTransformer as SpatialTransformer
+from .mv_attention import SPADTransformer as SpatialTransformer
+
 
 
-from util import exists
 from torch import autocast
 
 # dummy replace