diff --git a/Project_Model/Libs/Transformer/Classes/DeToken.py b/Project_Model/Libs/Transformer/Classes/DeToken.py
index c0b961e..1742374 100644
--- a/Project_Model/Libs/Transformer/Classes/DeToken.py
+++ b/Project_Model/Libs/Transformer/Classes/DeToken.py
@@ -14,6 +14,6 @@ class DeToken(torch.nn.Module):
         x = self.__linear(x)
 
         # 2) Go to logits
-        x = torch.softmax(x, 2)
+        # x = torch.softmax(x, 2)
 
         return x