Upload MemoryLLM

config.json

@@ -0,0 +1,57 @@
+{
+  "_name_or_path": "memoryllm-8b-chat",
+  "add_bos_embedding": true,
+  "add_decoder_lora": true,
+  "architectures": [
+    "MemoryLLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "bos_token_id": 128000,
+  "drop_memory_per_layer": true,
+  "eos_token_id": 128001,
+  "hidden_act": "silu",
+  "hidden_size": 4096,
+  "initializer_range": 0.02,
+  "intermediate_size": 14336,
+  "lora_config": {
+    "inference_mode": false,
+    "lora_alpha": 32,
+    "lora_dropout": 0.1,
+    "r": 16,
+    "target_modules": [
+      "q_proj",
+      "v_proj",
+      "k_proj",
+      "up_proj",
+      "down_proj",
+      "gate_proj"
+    ]
+  },
+  "max_length": 512,
+  "max_position_embeddings": 8192,
+  "max_seq_length": 2048,
+  "mlp_bias": false,
+  "model_type": "memoryllm",
+  "new_memory_embedding_fullset": true,
+  "num_attention_heads": 32,
+  "num_blocks": 50,
+  "num_hidden_layers": 32,
+  "num_key_value_heads": 8,
+  "num_memory_tokens": 12800,
+  "num_tokens": 256,
+  "pad_token_id": 128001,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": {
+    "factor": 1.8125,
+    "type": "linear"
+  },
+  "rope_theta": 500000.0,
+  "shrink_to_one_embedding": true,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float32",
+  "transformers_version": "4.43.0.dev0",
+  "use_cache": true,
+  "vocab_size": 128256
+}

generation_config.json

@@ -0,0 +1,9 @@
+{
+  "bos_token_id": 128000,
+  "do_sample": true,
+  "eos_token_id": 128001,
+  "max_length": 4096,
+  "temperature": 0.6,
+  "top_p": 0.9,
+  "transformers_version": "4.43.0.dev0"
+}

model-00001-of-00008.safetensors

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

model-00002-of-00008.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:496767754fbb806ed5666607a0ae60daab5a7173208a62e06154a06ac14116f1
+size 4917689241

model-00003-of-00008.safetensors

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

model-00004-of-00008.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:13b18147bef5aa85bc9bcc820893862c9803d1da156b71019c19b3bd2e97df57
+size 4816854808

model-00005-of-00008.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:58733b1be1d8b93ec09c833ebd28e70a9e4ef3f2e742215b7e68cf5edfab294a
+size 4883929264

model-00006-of-00008.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:50b8f69fc77bb9b0dee767b56100d10bacea05882cb15e4aa92edf9949034a86
+size 4816854856

model-00007-of-00008.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:410de571515c3fd957296da3d6a1e1579820f575fef227ce5517496f5d4e162c
+size 4883929264

model-00008-of-00008.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9a90c176b152a64e78d5b756b9562ac1a251009ce70bb766b3de43f5232aad9f
+size 3220526344