Upload 5 files

lexical_model.py

@@ -1,3 +1,4 @@
+# file: lexical_model.py
 import torch
 import torch.nn as nn
 from transformers import PreTrainedModel, PretrainedConfig
@@ -24,6 +25,7 @@ class LexicalHFModel(PreTrainedModel):
         super().__init__(config)
         self.config = config
         
+        # We use the exact name 'embedding' so it matches your .pth state_dict keys
         self.embedding = nn.Embedding(
             config.vocab_size, 
             config.embed_dim, 

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:edbee956054f61d9296c5668d71a320d02268f98369a1e75f8cf5b02130cd2ed
+oid sha256:ed3741f068bcec9b34e9d299dac8ccca5c55d5df5f5cdb4bc6f39113214c4310
 size 66363520