From 109ad9f36b352601b6a2254fa39677f8b58a4540 Mon Sep 17 00:00:00 2001
From: Christian Risi <75698846+CnF-Gris@users.noreply.github.com>
Date: Tue, 7 Oct 2025 16:36:59 +0200
Subject: [PATCH] Changed Imports

---
 Project_Model/Libs/Transformer/Classes/__init__.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/Project_Model/Libs/Transformer/Classes/__init__.py b/Project_Model/Libs/Transformer/Classes/__init__.py
index 837ea82..b4507f1 100644
--- a/Project_Model/Libs/Transformer/Classes/__init__.py
+++ b/Project_Model/Libs/Transformer/Classes/__init__.py
@@ -1,15 +1,16 @@
 from .Decoder import Decoder
 from .Encoder import Encoder
 from .FeedForwardNetwork import FeedForwardNetwork
-from .MultiHeadAttention import MultiheadAttention
+# from .MultiHeadAttention import MultiheadAttention
 from .TorchMultiHeadAttention import TorchMultiHeadAttention
 from .SpannedMasker import SpannedMasker
+from .DeToken import DeToken
 
 __all__ = [
     "Decoder",
     "Encoder",
     "FeedForwardNetwork",
-    "MultiheadAttention",
     "TorchMultiHeadAttention",
-    "SpannedMasker"
+    "SpannedMasker",
+    "DeToken"
 ]
\ No newline at end of file