0.1.0

TEST/0.1.0-base/config.json

@@ -0,0 +1,33 @@
+{
+  "_name_or_path": "/fsx-llm/lblecher/checkpoints/nougat/small/20230426_125023",
+  "align_long_axis": false,
+  "architectures": [
+    "NougatModel"
+  ],
+  "decoder_layer": 4,
+  "embed_dim": 128,
+  "encoder_layer": [
+    2,
+    2,
+    14,
+    2
+  ],
+  "hidden_dimension": 1024,
+  "input_size": [
+    896,
+    672
+  ],
+  "max_length": 3584,
+  "max_position_embeddings": 3584,
+  "model_type": "nougat",
+  "num_heads": [
+    4,
+    8,
+    16,
+    32
+  ],
+  "patch_size": 4,
+  "torch_dtype": "float32",
+  "transformers_version": "4.25.1",
+  "window_size": 7
+}

TEST/0.1.0-base/special_tokens_map.json

@@ -0,0 +1,6 @@
+{
+  "bos_token": "<s>",
+  "eos_token": "</s>",
+  "pad_token": "<pad>",
+  "unk_token": "<unk>"
+}

TEST/0.1.0-base/tokenizer_config.json

@@ -0,0 +1,4 @@
+{
+  "model_max_length": 1000000000000000019884624838656,
+  "tokenizer_class": "PreTrainedTokenizerFast"
+}