config.json

{
  "WEIGHT_DIMENSIONS": {
    "edge_index.dtype": "torch.int64",
    "edge_index.shape": [
      2,
      69125
    ],
    "edge_type.dtype": "torch.int64",
    "edge_type.shape": [
      69125
    ]
  },
  "_name_or_path": "./save/unicrs_rec/inspired/model_best",
  "architectures": [
    "UnicrsRec"
  ],
  "kgprompt_config": {
    "hidden_size": 768,
    "kg_info": {
      "num_entities": 17321,
      "num_relations": 13,
      "pad_entity_id": 17320
    },
    "n_block": 2,
    "n_head": 12,
    "n_layer": 12,
    "n_prefix_rec": 10,
    "num_bases": 8,
    "num_tokens": 50266,
    "token_hidden_size": 768
  },
  "num_tokens": 50259,
  "pad_token_id": 50257,
  "pretrained_model": "microsoft/DialoGPT-small",
  "torch_dtype": "float32",
  "transformers_version": "4.29.2"
}