fix relative import

Author: atiorh

ane_transformers/_version.py

@@ -1 +1 @@
-__version__ = "0.1.2"
+__version__ = "0.1.3"

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):

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):

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):

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):