Upload Concept Memory base model

README.md

@@ -1,3 +1,68 @@
 ---
-license: apache-2.0
+library_name: concept-memory-model
+tags:
+  - text-classification
+  - multi-task
+  - pytorch
+  - transformers
+datasets:
+  - mteb/banking77
+  - nyu-mll/glue
+  - google/boolq
 ---
+
+# Concept Memory typed judgment base model
+
+This repository contains the task-fine-tuned Concept Memory checkpoint for
+criteria-conditioned Choice, ordinal Score, and Yes/No (Noul) judgments. It is
+not a language model pretrained from scratch. Its text encoder is
+`sentence-transformers/all-mpnet-base-v2`.
+
+## Files
+
+- `model.pt`: PyTorch state dictionary and training metadata.
+- `config.json`: encoder, task, and training run metadata.
+
+## Load the checkpoint
+
+Install this project's dependencies, then load it with the repository's model
+class:
+
+```python
+import torch
+from huggingface_hub import hf_hub_download
+from concept_memory import ConceptMemoryModel
+
+repo_id = "Syigen/rgcm"
+checkpoint = hf_hub_download(repo_id, "model.pt")
+payload = torch.load(checkpoint, map_location="cpu", weights_only=False)
+model = ConceptMemoryModel.from_pretrained(
+    payload["encoder"],
+    decision_size=payload["decision_size"],
+    max_length=payload["max_length"],
+)
+model.load_state_dict(payload["model_state_dict"])
+model.eval()
+```
+
+The Hub checkpoint is a PyTorch artifact for this project's Python package; it
+is not directly compatible with `transformers.AutoModel.from_pretrained`.
+
+## Training metadata
+
+{
+  "datasets": [
+    "mteb/banking77",
+    "nyu-mll/glue:sst2",
+    "google/boolq"
+  ],
+  "task_types": [
+    "choice",
+    "score",
+    "noul"
+  ],
+  "encoder": "sentence-transformers/all-mpnet-base-v2",
+  "epochs": 10,
+  "batch_size": 16,
+  "seed": 42
+}

config.json

@@ -0,0 +1,16 @@
+{
+  "datasets": [
+    "mteb/banking77",
+    "nyu-mll/glue:sst2",
+    "google/boolq"
+  ],
+  "task_types": [
+    "choice",
+    "score",
+    "noul"
+  ],
+  "encoder": "sentence-transformers/all-mpnet-base-v2",
+  "epochs": 10,
+  "batch_size": 16,
+  "seed": 42
+}

model.pt

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