Fixed several bugs for task 4

2025-10-12 16:30:30 +02:00
parent e0f8a36aa5
commit 07130ff489
5 changed files with 186 additions and 9 deletions
--- a/Project_Model/Libs/TransformerUtils/model_utils.py
+++ b/Project_Model/Libs/TransformerUtils/model_utils.py
@@ -10,8 +10,7 @@ def decompose_nano_socrates(
 ) -> tuple[TrainingModel, NanoSocratEncoder, NanoSocraDecoder]:

    encoder_pieces, decoder_pieces = model.take_pieces()
-    encoder_embedder, encoder = encoder_pieces
-    encoder_detokener = DeToken(embedding_size, vocabulary_size)
+    encoder_embedder, encoder, encoder_detokener = encoder_pieces
    decoder_embedder, decoder, decoder_detokener = decoder_pieces

    return (