Update evo_model.py

evo_model.py

@@ -1,11 +1,13 @@
-import torch.nn as nn
+# evo_model.py — EvoDecoderModel matching saved checkpoint
 import torch
+import torch.nn as nn
 
 class EvoDecoderModel(nn.Module):
     def __init__(self, vocab_size, d_model=512, nhead=8, num_layers=6, dim_feedforward=2048, dropout=0.1, max_len=128):
         super().__init__()
         self.token_embed = nn.Embedding(vocab_size, d_model)
-        self.pos_encoder = nn.Parameter(torch.zeros(1, max_len, d_model))  # match [1, 128, 512]
+        self.pos_encoder = nn.Parameter(torch.zeros(1, max_len, d_model))  # matches saved shape [1, 128, 512]
+
         decoder_layer = nn.TransformerDecoderLayer(
             d_model=d_model,
             nhead=nhead,