Upload 4 files

enigma_module.py

@@ -17,6 +17,7 @@ class EnigmaModel(PreTrainedModel):
         super().__init__(config)
         self.embedding = nn.Embedding(config.vocab_size, config.hidden_size)
         self.linear = nn.Linear(config.hidden_size, config.hidden_size)
+        self.post_init()
         
     def forward(self, input_ids):
         x = self.embedding(input_ids)

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c971f99b962157c9dc4032334429f8fb6c59afb35daa8ed563ad7c867e036f38
+oid sha256:78e404d4dc8e88bc893ccc0b188a5e095da978c676a4cde1c1a7bce429eb70ed
 size 2626336