Update model.py

model.py

@@ -46,7 +46,7 @@ class FleshkaTabularTransformer(PreTrainedModel):
         # x: [batch_size, input_dim]
         out = list()
         for nx in lx:
-            x = tensor(self._normalize(nx), dtype=float32).unsqueeze(0)
+            x = tensor(self._normalize(nx), dtype=float32).unsqueeze(0).to(self.device)
             x = self.input_proj(x)           # [batch_size, d_model]
             x = x.unsqueeze(1)               # [batch_size, 1, d_model] (добавляем seq_len=1)
             x = self.transformer(x)          # [batch_size, 1, d_model]