Upload lambda-360m pretrained model

model.pth

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

model_config.json

@@ -0,0 +1 @@
+{"max_len": 1024, "d_model": 960, "num_layers": 32, "num_heads": 15, "num_kv_heads": 5, "d_ff": 4096, "learning_rate": 0.0002, "batch_size": 48, "gradient_accumulation_steps": 2, "devices": "auto", "device_count": 2, "global_batch_size": 96, "effective_batch_size": 96, "global_effective_batch_size": 192, "lr_schedule": "warmup_cosine", "lr_warmup_steps": 2000, "min_learning_rate": 4e-05, "min_learning_rate_ratio": 0.2, "pad_token_id": 0, "bos_token_id": 2, "eos_token_id": 3, "train_dataset_signature": "4e423c65d9d18d90", "validation_dataset_signature": "571c26264739139a", "train_dataset_case": {"name": "lambda-corpus-train", "genre": "mixed", "language": "ja", "dataset_path": "KeisukeMiyamoto/lambda-corpus", "config_name": "default", "split": "train", "text_column": "text"}, "validation_dataset_case": {"name": "lambda-corpus-validation", "genre": "mixed", "language": "ja", "dataset_path": "KeisukeMiyamoto/lambda-corpus", "config_name": "default", "split": "validation", "text_column": "text"}, "validation_cache_path": "models/lambda-360m/validation-cache-571c26264739139a-bucket-packing-v1-len1024-samples6144.pt", "validation_sample_count": 6144, "packing_version": "bucket-packing-v1", "shuffle_buffer_size": 10000, "base_shuffle_seed": 17, "shuffle_seed": 17, "trained_steps": 48000, "training_precision": "fp8", "float8_recipe": "tensorwise"}

special_tokens_map.json

@@ -0,0 +1,17 @@
+{
+  "pad_token": "|<pad>|",
+  "unk_token": "|<unknown>|",
+  "bos_token": "|<bos>|",
+  "eos_token": "|<eos>|",
+  "sep_token": "|<sep>|",
+  "cls_token": "|<cls>|",
+  "mask_token": "|<mask>|",
+  "extra_special_tokens": [
+    "|<system>|",
+    "|<user>|",
+    "|<assistant>|",
+    "|<thinking>|",
+    "|<end_of_thinking>|",
+    "|<end_of_turn>|"
+  ]
+}

tokenizer_config.json

@@ -0,0 +1,20 @@
+{
+  "backend": "tokenizers",
+  "bos_token": "|<bos>|",
+  "cls_token": "|<cls>|",
+  "eos_token": "|<eos>|",
+  "extra_special_tokens": [
+    "|<system>|",
+    "|<user>|",
+    "|<assistant>|",
+    "|<thinking>|",
+    "|<end_of_thinking>|",
+    "|<end_of_turn>|"
+  ],
+  "mask_token": "|<mask>|",
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "|<pad>|",
+  "sep_token": "|<sep>|",
+  "tokenizer_class": "TokenizersBackend",
+  "unk_token": "|<unknown>|"
+}