Update model.py

model.py

@@ -45,7 +45,7 @@ class FleshkaTabularTransformer(PreTrainedModel):
         out = []
         for nx in lx:
             # Убедимся, что входные данные в float16
-            x = tensor(self._normalize(nx), dtype=self.input_proj.dtype).unsqueeze(0).to(self.device)
+            x = tensor(self._normalize(nx), dtype=self.input_proj.weight.dtype).unsqueeze(0).to(self.device)
             x = self.input_proj(x)
             x = x.unsqueeze(1)
             x = self.transformer(x)