Upload 3 files

config.json

@@ -0,0 +1,12 @@
+{
+    "architectures": [
+        "TRMForConditionalGeneration"
+    ],
+    "hidden_size": 512,
+    "num_hidden_layers": 2,
+    "num_attention_heads": 8,
+    "recursive_steps": 16,
+    "model_type": "trm",
+    "torch_dtype": "float32",
+    "transformers_version": "4.40.0"
+}

model.safetensors

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

tokenizer_config.json

@@ -0,0 +1 @@
+{"tokenizer_class": "PreTrainedTokenizer"}