fix relative import

KE7 · Aug 9, 2022 · da64000 · da64000
1 parent 4b37184
commit da64000
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/ane_transformers/_version.py b/ane_transformers/_version.py
@@ -1 +1 @@
-__version__ = "0.1.2"
+__version__ = "0.1.3"
diff --git a/ane_transformers/reference/decoder.py b/ane_transformers/reference/decoder.py
@@ -12,7 +12,7 @@
 import torch
 import torch.nn as nn
 
-from layer_norm import LayerNormANE
+from .layer_norm import LayerNormANE
 
 
 class TransformerDecoder(nn.Module):

diff --git a/ane_transformers/reference/ffn.py b/ane_transformers/reference/ffn.py
@@ -6,7 +6,7 @@
 import torch
 import torch.nn as nn
 
-from layer_norm import LayerNormANE
+from .layer_norm import LayerNormANE
 
 
 class FFN(nn.Module):

diff --git a/ane_transformers/reference/multihead_attention.py b/ane_transformers/reference/multihead_attention.py
@@ -6,7 +6,7 @@
 import torch
 import torch.nn as nn
 
-from layer_norm import LayerNormANE
+from .layer_norm import LayerNormANE
 
 
 class MultiHeadAttention(nn.Module):

diff --git a/ane_transformers/reference/transformer.py b/ane_transformers/reference/transformer.py
@@ -12,7 +12,7 @@
 import torch
 import torch.nn as nn
 
-import encoder, decoder, multihead_attention, ffn
+from . import encoder, decoder, multihead_attention, ffn
 
 
 class AppleNeuralEngineTransformer(nn.Module):