Upload UnicrsGen

config.json

@@ -0,0 +1,39 @@
+{
+  "WEIGHT_DIMENSIONS": {
+    "edge_index.dtype": "torch.int64",
+    "edge_index.shape": [
+      2,
+      186685
+    ],
+    "edge_type.dtype": "torch.int64",
+    "edge_type.shape": [
+      186685
+    ]
+  },
+  "_name_or_path": "RecWizard/unicrs-gen-redial",
+  "architectures": [
+    "UnicrsGen"
+  ],
+  "kgprompt_config": {
+    "hidden_size": 768,
+    "kg_info": {
+      "num_entities": 31162,
+      "num_relations": 25,
+      "pad_entity_id": 31161
+    },
+    "n_block": 2,
+    "n_head": 12,
+    "n_layer": 12,
+    "n_prefix_conv": 20,
+    "n_prefix_rec": null,
+    "num_bases": 8,
+    "num_tokens": 50266,
+    "token_hidden_size": 768
+  },
+  "max_gen_len": 50,
+  "num_tokens": 50259,
+  "pad_token_id": 50257,
+  "pretrained_model": "microsoft/DialoGPT-small",
+  "torch_dtype": "float32",
+  "transformers_version": "4.33.2"
+}

generation_config.json

@@ -0,0 +1,5 @@
+{
+  "_from_model_config": true,
+  "pad_token_id": 50257,
+  "transformers_version": "4.33.2"
+}

pytorch_model.bin

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