ysharma HF staff commited on
Commit
971671e
1 Parent(s): 55451b9
Files changed (1) hide show
  1. openaimodel.py +2 -2
openaimodel.py CHANGED
@@ -8,7 +8,7 @@ import torch as th
8
  import torch.nn as nn
9
  import torch.nn.functional as F
10
 
11
- from ldm.modules.diffusionmodules.util import (
12
  checkpoint,
13
  conv_nd,
14
  linear,
@@ -17,7 +17,7 @@ from ldm.modules.diffusionmodules.util import (
17
  normalization,
18
  timestep_embedding,
19
  )
20
- from ldm.modules.attention import SpatialTransformer
21
 
22
 
23
  # dummy replace
 
8
  import torch.nn as nn
9
  import torch.nn.functional as F
10
 
11
+ from util2 import (
12
  checkpoint,
13
  conv_nd,
14
  linear,
 
17
  normalization,
18
  timestep_embedding,
19
  )
20
+ from attention import SpatialTransformer
21
 
22
 
23
  # dummy replace