ysharma HF staff commited on
Commit
55451b9
1 Parent(s): 8ffd87e
Files changed (1) hide show
  1. modules.py +1 -1
modules.py CHANGED
@@ -2,7 +2,7 @@ import torch
2
  import torch.nn as nn
3
  from functools import partial
4
 
5
- from ldm.modules.x_transformer import Encoder, TransformerWrapper # TODO: can we directly rely on lucidrains code and simply add this as a reuirement? --> test
6
 
7
 
8
  class AbstractEncoder(nn.Module):
 
2
  import torch.nn as nn
3
  from functools import partial
4
 
5
+ from x_transformer import Encoder, TransformerWrapper # TODO: can we directly rely on lucidrains code and simply add this as a reuirement? --> test
6
 
7
 
8
  class AbstractEncoder(nn.Module):