Upload folder using huggingface_hub

config.json

@@ -0,0 +1,21 @@
+{
+  "architectures": [
+    "MemoryCausalLM"
+  ],
+  "auto_map": {
+    "AutoConfig": "model.MemoryConfig",
+    "AutoModelForCausalLM": "model.MemoryCausalLM"
+  },
+  "base_model": "gpt2-xl",
+  "dtype": "bfloat16",
+  "extractor_type": "next-token",
+  "hidden_size": 1600,
+  "interpreter_type": "projection",
+  "max_length": 1024,
+  "max_memory_size": 65536,
+  "num_layers": 23,
+  "reader_type": "inverse",
+  "reader_update_discount": 0.9,
+  "reader_update_size": 8192,
+  "transformers_version": "4.57.1"
+}

model.py

@@ -0,0 +1,6 @@
+from llm import MemoryConfig, MemoryCausalLM
+
+__all__ = [
+    "MemoryConfig",
+    "MemoryCausalLM",
+]

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8edcf8c9f8d2aebd83191bc63f5a4994a93087f6b0a2431c72033c593303fc05
+size 3233199968

special_tokens_map.json

@@ -0,0 +1,23 @@
+{
+  "bos_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer_config.json

@@ -0,0 +1,20 @@
+{
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "50256": {
+      "content": "<|endoftext|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "bos_token": "<|endoftext|>",
+  "clean_up_tokenization_spaces": true,
+  "eos_token": "<|endoftext|>",
+  "extra_special_tokens": {},
+  "model_max_length": 1024,
+  "tokenizer_class": "GPT2Tokenizer",
+  "unk_token": "<|endoftext|>"
+}