Update model.py

model.py

@@ -1,5 +1,6 @@
 from transformers import PreTrainedModel
 import torch.nn as nn
+import numpy as np
 from torch import tensor, float32
 from torch.nn import TransformerEncoder, TransformerEncoderLayer
 from .model_config import TabularTransformerConfig
@@ -33,7 +34,7 @@ class FleshkaTabularTransformer(PreTrainedModel):
         self._init_weights()
 
     def _normalize(self, flist):
-        return (flist - self.config.mean) / (self.config.std + 1e-8)
+        return (flist - np.array(self.config.mean)) / (np.array(self.config.std) + 1e-8)
     def _init_weights(self):
         for module in self.modules():
             if isinstance(module, nn.Linear):